Update app.py

app.py

@@ -21,9 +21,10 @@ model.to(device)
 class ChatBot:
     def __init__(self):
         self.history = []
-
     def predict(self, user_input, system_prompt="You are an expert medical analyst:"):
-        prompt = [{'role': 'user', 'content': user_input}, {'role': 'system', 'content': system_prompt}]
+        formatted_prompt = f"{user_input}\n{system_prompt}:"
+        prompt = [{'role': 'user', 'content': formatted_prompt}]
+        prompt = [{'role': 'user', 'content': user_input "\n" system_prompt}]
         inputs = tokenizer.apply_chat_template(
             prompt,
             add_generation_prompt=True,