Heiko Hotz commited on
Commit
fca3b2e
1 Parent(s): 70a9509

initial commit

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -132,7 +132,7 @@ if st.button('Analyze'):
132
  print('getting predictions')
133
  with st.spinner(text='Analysis in progress...'):
134
  predictions = run_prediction([question], paragraph, 'akdeniz27/roberta-base-cuad',
135
- n_best_size=int(number_results))
136
  if predictions['0'] == "":
137
  answer = 'No answer found in document'
138
  else:
 
132
  print('getting predictions')
133
  with st.spinner(text='Analysis in progress...'):
134
  predictions = run_prediction([question], paragraph, 'akdeniz27/roberta-base-cuad',
135
+ n_best_size=10)
136
  if predictions['0'] == "":
137
  answer = 'No answer found in document'
138
  else: