Update app.py

app.py

@@ -24,8 +24,12 @@ def set_custom_prompt():
     """
     Prompt template for QA retrieval for vector stores
     """
-    prompt = PromptTemplate(template = custom_prompt_template,
-                            input_variables = ['context','question'])
+    # prompt = PromptTemplate(template = custom_prompt_template,
+    #                         input_variables = ['context','question'])
+    
+    # return prompt
+    prompt = PromptTemplate(template=custom_prompt_template,
+                            input_variables=['context', 'question'])  # Ensure this matches expected inputs
     
     return prompt