EncephalitisGPT / README.md
Santosh-Gupta's picture
Update README.md
888f89d
|
Raw
History Blame Contribute Delete
667 Bytes
metadata
license: mit

The training of model used Microsoft's bioGPT as a base model, and then finetuned in a dataset of over 370k encephalitis question and answers

https://huggingface.co/datasets/Santosh-Gupta/EncephalitisQuestionAnswering

To use

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)