Update app.py

app.py

@@ -25,8 +25,8 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],templat
 def predict(message, history):
     input_prompt = QA_CHAIN_PROMPT.format(question=message, context=history)
     result = llm.generate([input_prompt])
-    print(result)  # Add this line to inspect the result
-    # ai_msg = ... (update this line based on the result structure)
+    print(result)
+    ai_msg = result[0].generations[0].text
     return ai_msg