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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ def main():
149
 
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(answer_start_scores.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'])
 
149
 
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'])