Update app.py

app.py

@@ -24,7 +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])  # Pass a list of strings
+    ai_msg = llm.generate([input_prompt])[0]  # Return only the first generation
     return ai_msg
 
+
+
 gr.ChatInterface(predict).launch()