Update app.py

app.py

@@ -158,9 +158,9 @@ def answer_question(context, question):
             {"role":"system", "content": SYSTEM_MESSAGE},
             {"role": "user", "content": f"Context: {context}\n Question: {question}"}
         ]
-        
+        logger.info(response[])
         response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
-        return response[1]['content']
+        return response[1]
     except Exception as e:
         logger.error(f"Q&A failed: {str(e)}")
         return f"Error occurred during Q&A process: {str(e)}"