Update app.py

app.py

@@ -30,7 +30,9 @@ def askme(symptoms, question):
     # Remove system messages and content    
     # Extract and return the generated text, removing the prompt
     # Extract only the assistant's response
-    answer = response_text.split('<|im_start|>assistant')[-1].strip()  
+
+    # Extract only the assistant's response
+    answer = response_text.split('<|im_start|>assistant')[1].split('<|im_end|>')[0].strip()  
     return  answer