Update app.py

app.py

@@ -33,8 +33,10 @@ def askme(symptoms, question):
     response_text = tokenizer.batch_decode(outputs)[0].strip()
     # Remove system messages and content
     response_text = response_text.replace(sys_message, "").replace(content, "").strip()
+    # Extract only the assistant's response
+    assistant_response = response_text.split("assistant")[1].strip()
 
-    return response_text
+    return assistant_response
 
 # Example usage
 symptoms = '''\