Heiko Hotz commited on
Commit
e96d307
1 Parent(s): 21f7f52
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -134,7 +134,7 @@ if st.button('Analyze'):
134
  print('getting predictions')
135
  with st.spinner(text='Analysis in progress...'):
136
  predictions = run_prediction([question], paragraph, 'marshmellow77/roberta-base-cuad',
137
- n_best_size=10)
138
  answer = ""
139
  if predictions['0'] == "":
140
  answer = 'No answer found in document'
 
134
  print('getting predictions')
135
  with st.spinner(text='Analysis in progress...'):
136
  predictions = run_prediction([question], paragraph, 'marshmellow77/roberta-base-cuad',
137
+ n_best_size=5)
138
  answer = ""
139
  if predictions['0'] == "":
140
  answer = 'No answer found in document'