Update app.py
Browse files
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(
|
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 |
|