Spaces:
Build error
Build error
Heiko Hotz
commited on
Commit
•
70a9509
1
Parent(s):
0a89e6e
initial commit
Browse files- predict.py +1 -1
predict.py
CHANGED
|
@@ -102,7 +102,7 @@ def run_prediction(question_texts, context_text, model_path, n_best_size=1):
|
|
| 102 |
print(all_results)
|
| 103 |
|
| 104 |
output_nbest_file = None
|
| 105 |
-
if n_best_size
|
| 106 |
output_nbest_file = "nbest.json"
|
| 107 |
|
| 108 |
timer = time.time()
|
|
|
|
| 102 |
print(all_results)
|
| 103 |
|
| 104 |
output_nbest_file = None
|
| 105 |
+
if n_best_size > 1:
|
| 106 |
output_nbest_file = "nbest.json"
|
| 107 |
|
| 108 |
timer = time.time()
|