Update README.md

README.md

@@ -54,7 +54,17 @@ import transformers
 import torch
 
 model = "sethuiyer/Dr_Samantha_7b_mistral"
-messages = [{"role": "user", "content": "What is a large language model?"}]
+ask_samantha = '''
+Symptoms:
+Dizziness, headache and nausea.
+
+What is the differnetial diagnosis?
+'''
+
+messages = [{"role": "system", "content": '''You are Doctor Samantha, a virtual AI doctor known for your friendly and approachable demeanor, 
+combined with a deep expertise in the medical field. You're here to provide professional, empathetic, and knowledgeable advice on health-related inquiries.
+You'll also provide differential diagnosis. If you're unsure about any information, Don't share false information.'''},
+{"role": "user", "content": f"{ask_samantha}"}]
 
 tokenizer = AutoTokenizer.from_pretrained(model)
 prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
@@ -67,4 +77,15 @@ pipeline = transformers.pipeline(
 
 outputs = pipeline(prompt, max_new_tokens=256, do_sample=True, temperature=0.7, top_k=50, top_p=0.95)
 print(outputs[0]["generated_text"])
+```
+```text
+Dizziness, headache and nausea can be caused by a variety of conditions, including:
+
+Vertigo: A sensation of spinning or dizziness that can be caused by problems with the inner ear or brain.
+Migraine: A type of headache that can cause throbbing pain, sensitivity to light and sound, and nausea.
+Concussion: A type of traumatic brain injury that can cause dizziness, headache, and nausea.
+Dehydration: A lack of fluids in the body can cause dizziness, headache, and nausea.
+Low blood sugar: A drop in blood sugar levels can cause dizziness, headache, and nausea.
+
+It's important to consult with a healthcare professional for a proper diagnosis and treatment plan.
 ```