Kr08 commited on
Commit
a900380
1 Parent(s): 4c6fb9e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -156,7 +156,7 @@ def answer_question(context, question):
156
  messages = [
157
  # {"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
158
  {"role":"system", "content": SYSTEM_MESSAGE},
159
- {"role": "user", "content": f"Context: {text}\n\nQuestion: {question}"}
160
  ]
161
 
162
  response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']
 
156
  messages = [
157
  # {"role": "system", "content": "You are a helpful assistant who can answer questions based on the given context."},
158
  {"role":"system", "content": SYSTEM_MESSAGE},
159
+ {"role": "user", "content": f"Context: {context}\n\nQuestion: {question}"}
160
  ]
161
 
162
  response = qa_pipeline(messages, max_new_tokens=256)[0]['generated_text']