Update app.py

app.py

@@ -160,7 +160,7 @@ def answer_question(context, question):
         ]
         response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
         logger.info(response)
-        return response[1]
+        return response[-1]['content']
     except Exception as e:
         logger.error(f"Q&A failed: {str(e)}")
         return f"Error occurred during Q&A process: {str(e)}"