Lisibonny commited on
Commit
ec74d09
1 Parent(s): d403bd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -150,7 +150,7 @@ def main():
150
  answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
151
 
152
  all_results.loc[i] = answer, max(outputs.start_logits.numpy()[0]), 0, 0
153
- st.write(df_answer.sort_values(by=['score']).tail(10))
154
 
155
  #outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
156
 
 
150
  answer = tokenizer.convert_tokens_to_string(tokenizer.convert_ids_to_tokens(input_ids[answer_start:answer_end]))
151
 
152
  all_results.loc[i] = answer, max(outputs.start_logits.numpy()[0]), 0, 0
153
+ st.write(all_results.sort_values(by=['score']).tail(10))
154
 
155
  #outputs = qa_model(input_ids=inputs['input_ids'], attention_mask=inputs['attention_mask'])
156