Update app.py

app.py

@@ -24,9 +24,9 @@ QA_CHAIN_PROMPT = PromptTemplate(input_variables=["context", "question"],templat
 
 def predict(message, history):
     input_prompt = QA_CHAIN_PROMPT.format(question=message, context=history)
-    ai_msg = llm.generate([input_prompt])[0]  # Return only the first generation
+    result = llm.generate([input_prompt])
+    ai_msg = result[0].generations[0].text
     return ai_msg
 
 
-
 gr.ChatInterface(predict).launch()