Update app.py

app.py

@@ -154,13 +154,13 @@ def answer_question(context, question):
             return "Please enter your Question"
 
         messages = [
-            {"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
-            # {"role":"system", "content": SYSTEM_MESSAGE},
+            # {"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
+            {"role":"system", "content": SYSTEM_MESSAGE},
             {"role": "user", "content": f"Context: {context}\n Question: {question}"}
         ]
         
         response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
-        return response
+        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)}"