Update app.py

app.py

@@ -27,44 +27,16 @@ def askme(symptoms, question):
     inputs = tokenizer(prompt, return_tensors="pt").to(device)
     outputs = model.generate(**inputs, max_new_tokens=300, use_cache=True)
     response_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0].strip()
-
-    #start_idx = response_text.find("<|im_start|>assistant")
-    #end_idx = response_text.find("<|im_end|>", start_idx)
-    #assistant_response = response_text[start_idx + len("<|im_start|>assistant"):end_idx]
-
-    #return assistant_response.split(". ")[0] + "
     return response_text
 
 
-
-
-
-
-def askmeold(symptoms, question):
-    custom_template = [
-        {"role": "system", "content": "You are an AI Medical Assistant trained on a vast dataset of health information. Please be thorough and provide an informative answer. If you don't know the answer to a specific medical inquiry, advise seeking professional help."},
-        {"role": "user", "content": "Symptoms: {symptoms}\nQuestion: {question}\n"},
-        {"role": "assistant", "content": "{assistant_response}\n"}
-    ]
-
-    sys_message = custom_template[0]["content"]
-    content = custom_template[1]["content"].format(symptoms=symptoms, question=question)
-    messages = [{"role": "system", "content": sys_message}, {"role": "user", "content": content}]
-
-    prompt = tokenizer.apply_chat_template(messages, template=custom_template, tokenize=False, add_generation_prompt=True)
-    inputs = tokenizer(prompt, return_tensors="pt").to(device)  # Ensure inputs are on CUDA device
-    outputs = model.generate(**inputs, max_new_tokens=300, use_cache=True)
-    response_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0].strip() 
-
-    # Extract only the assistant's response
-    start_idx = response_text.find("<|im_start|>assistant")
-    end_idx = response_text.find("<|im_end|>", start_idx)
-    assistant_response = response_text[start_idx + len("<|im_start|>assistant"):end_idx]
-
-    # Return only one answer
-    answers = assistant_response.split(". ")
-    return answers[0] + "."
-    
+examples = [
+    ["headache", "What are the possible causes of a headache?"],
+    ["fever", "How can I treat a fever at home?"],
+    ["cough", "What are the symptoms of a cough?"],
+    ["chest pain", "What are the possible causes of chest pain?"],
+]
+   
 
 # Example usage
 symptoms = '''\
@@ -128,12 +100,7 @@ Developed by Ruslan Magana. Visit [https://ruslanmv.com/](https://ruslanmv.com/)
 
 
 
-examples = [
-    ["headache", "What are the possible causes of a headache?"],
-    ["fever", "How can I treat a fever at home?"],
-    ["cough", "What are the symptoms of a cough?"],
-    ["chest pain", "What are the possible causes of chest pain?"],
-]
+
 
 
 symptoms_input = gr.Textbox(label="Symptoms")