Update app.py

app.py

@@ -32,11 +32,9 @@ def askme(symptoms, question):
     outputs = model.generate(**inputs, max_new_tokens=200, use_cache=True)
     response_text = tokenizer.batch_decode(outputs)[0].strip()
     # Remove system messages and content
-    response_text = response_text.replace(sys_message, "").replace(content, "").strip()
+    #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 assistant_response
+    return response_text
 
 # Example usage
 symptoms = '''\