Update app.py

app.py

@@ -33,8 +33,8 @@ def askme(symptoms, question):
     response_text = tokenizer.batch_decode(outputs, skip_special_tokens=True)[0].strip()
 
     # Extract only the assistant's response
-    assistant_response = response_text.split("<|im_start|>assistant")[-1].strip()
-    return assistant_response
+    #assistant_response = response_text.split("<|im_start|>assistant")[-1].strip()
+    return response_text