ziyadbastaili commited on
Commit
90e227c
1 Parent(s): ba2367e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -3
app.py CHANGED
@@ -89,6 +89,10 @@ def run_prediction(context_text, question):
89
  result = SquadResult(unique_id, start_logits, end_logits)
90
  all_results.append(result)
91
 
 
 
 
 
92
  predictions = compute_predictions_logits(
93
  examples,
94
  features,
@@ -96,9 +100,9 @@ def run_prediction(context_text, question):
96
  n_best_size,
97
  max_answer_length,
98
  do_lower_case,
99
- False,
100
- False,
101
- False,
102
  False, # verbose_logging
103
  True, # version_2_with_negative
104
  null_score_diff_threshold,
 
89
  result = SquadResult(unique_id, start_logits, end_logits)
90
  all_results.append(result)
91
 
92
+ output_prediction_file = "predictions.json"
93
+ output_nbest_file = "nbest_predictions.json"
94
+ output_null_log_odds_file = "null_predictions.json"
95
+
96
  predictions = compute_predictions_logits(
97
  examples,
98
  features,
 
100
  n_best_size,
101
  max_answer_length,
102
  do_lower_case,
103
+ output_prediction_file,
104
+ output_nbest_file,
105
+ output_null_log_odds_file,
106
  False, # verbose_logging
107
  True, # version_2_with_negative
108
  null_score_diff_threshold,