Update app.py

app.py

@@ -30,12 +30,8 @@ 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()  
-    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]  
+    answer = response_text.split('<|im_start|>assistant')[-1].strip()  
+    return  answer