Update app.py

app.py

@@ -21,7 +21,7 @@ def askme(symptoms, question):
     You are an AI Medical Assistant trained on a vast dataset of health information. Please be thorough and
     provide an informative answer. If you don't know the answer to a specific medical inquiry, advise seeking professional help.
     '''
-    content = symptoms + " " + question
+    content = "symptoms:"+ symptoms + "question:" + question
     messages = [{"role": "system", "content": sys_message}, {"role": "user", "content": content}]
     prompt = tokenizer.apply_chat_template(messages, tokenize=False, add_generation_prompt=True)
     inputs = tokenizer(prompt, return_tensors="pt").to(device)  # Ensure inputs are on CUDA device
@@ -30,9 +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()
+    #answer = response_text.split('<|im_start|>assistant')[-1].strip()
     #answer =response_text.split("assistant")[1].strip().split("user")[0].strip()   
-    return answer
+    return response_text
 
 # Example usage
 symptoms = '''\