Views
No views yet
1from transformers import AutoModelForCausalLM, AutoTokenizer
2
3model = AutoModelForCausalLM.from_pretrained("vyshnaviii123/qwen-medical-qa")
4tokenizer = AutoTokenizer.from_pretrained("vyshnaviii123/qwen-medical-qa")
5
6input = tokenizer("Q: What helps with a sore throat?\nA:", return_tensors="pt")
7output = model.generate(**input, max_new_tokens=100)
8print(tokenizer.decode(output[0]))