alexpap commited on
Commit
4e300f9
1 Parent(s): 90fbbbe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -6
app.py CHANGED
@@ -170,19 +170,19 @@ elif menu == "Evaluation":
170
  Feel free to query the pre-trained QA-NLU model using the buttons below.
171
  ''')
172
 
173
- tokenizer = AutoTokenizer.from_pretrained("AmazonScience/qanlu")#, use_auth_token=True)
174
 
175
- model = AutoModelForQuestionAnswering.from_pretrained("AmazonScience/qanlu")#, use_auth_token=True)
176
 
177
  qa_pipeline = pipeline('question-answering', model=model, tokenizer=tokenizer)
178
 
179
  context = st.text_input(
180
- 'Please enter the context:',
181
- value="I want a cheap flight to Boston."
182
  )
183
  question = st.text_input(
184
- 'Please enter the question:',
185
- value="What is the destination?"
186
  )
187
 
188
 
 
170
  Feel free to query the pre-trained QA-NLU model using the buttons below.
171
  ''')
172
 
173
+ tokenizer = AutoTokenizer.from_pretrained("AmazonScience/qanlu")
174
 
175
+ model = AutoModelForQuestionAnswering.from_pretrained("AmazonScience/qanlu")
176
 
177
  qa_pipeline = pipeline('question-answering', model=model, tokenizer=tokenizer)
178
 
179
  context = st.text_input(
180
+ 'Please enter the context (remember to include "Yes. No. " in the beginning):',
181
+ value="Yes. No. I want a cheap flight to Boston."
182
  )
183
  question = st.text_input(
184
+ 'Please enter the intent question:',
185
+ value="Are they looking for a flight?"
186
  )
187
 
188