Views
No views yet
1from transformers import pipeline, AutoTokenizer
2
3model_path = "PradeepBodhi/BioMistral-7b_Fine-Tuned-QLoRA"
4tokenizer = AutoTokenizer.from_pretrained(model_path)
5pipe = pipeline("text-generation", model=model_path, tokenizer=tokenizer)
6
7prompt = tokenizer.apply_chat_template([
8 {"role": "user", "content": "I have a sore throat and fever. What should I do?"}
9], tokenize=False, add_generation_prompt=True)
10
11output = pipe(prompt, max_new_tokens=1024, do_sample=True, temperature=0.7)
12print(output[0]['generated_text'][len(prompt):].strip())