Update app.py

app.py

@@ -10,15 +10,10 @@ def generate_answer(query):
         model_kwargs={"temperature": 0.7, "max_length": 64, "max_new_tokens": 512}
     )
 
-    template = """Patient's Description: {patient_description}
-
-Doctor's Response: Thank you for sharing details about your headaches. Based on what you've described, it seems like you're managing your symptoms well by identifying triggers, taking naps, practicing yoga, and staying hydrated. It's important to continue these healthy habits.
-
-To further assist you, could you provide more insight into any specific patterns you've noticed with your headaches? For example, do they follow a particular schedule or worsen during certain times of the day? Additionally, have you experienced any other associated symptoms that might be relevant?
-
-Your proactive approach to managing stress and avoiding triggers is commendable. Let's work together to explore additional strategies that may enhance your overall well-being.
-"""
-    
+    template = """Question: {question}
+         
+    Answer: Let's think step by step."""
+            
     prompt = PromptTemplate(template=template, input_variables=["query"])
     llm_chain = LLMChain(prompt=prompt, llm=llm)
     result = llm_chain.run(query)