Views
No views yet
from transformers import pipeline, AutoTokenizer, AutoModel
model = AutoModel.from_pretrained('Santosh-Gupta/EncephalitisGPT')
tokenizer = AutoTokenizer.from_pretrained('Santosh-Gupta/EncephalitisGPT')
generator = pipeline('text-generation', model=model, tokenizer=tokenizer, device=0)
generator("What are causes of autoimmune encephalitis", max_length=1000, num_return_sequences=1, do_sample=True)