Views
No views yet
1from transformers import AutoTokenizer, AutoModelForCausalLM
2
3# Load model
4model = AutoModelForCausalLM.from_pretrained("riad-diseases-gpt", trust_remote_code=True)
5tokenizer = AutoTokenizer.from_pretrained("riad-diseases-gpt", trust_remote_code=True)
6
7# Generate answer
8question = "What is dengue fever?"
9answer = model.generate_text(tokenizer, question)
10print(answer)1@misc{riad_disease_gpt,
2 author = {Your Name},
3 title = {Riad Disease GPT},
4 year = {2024},
5 publisher = {Hugging Face},
6 howpublished = {\url{https://huggingface.co/riad-diseases-gpt}}
7}