Update Cuad_others.py
Browse files- Cuad_others.py +1 -1
Cuad_others.py
CHANGED
|
@@ -52,7 +52,7 @@ def quad(query,file):
|
|
| 52 |
questions_short = load_questions_short()
|
| 53 |
if (not len(paragraph)==0) and not (len(query)==0):
|
| 54 |
print('getting predictions')
|
| 55 |
-
predictions = run_prediction([query], paragraph, '
|
| 56 |
answer = ""
|
| 57 |
answer_p=""
|
| 58 |
if predictions['0'] == "":
|
|
|
|
| 52 |
questions_short = load_questions_short()
|
| 53 |
if (not len(paragraph)==0) and not (len(query)==0):
|
| 54 |
print('getting predictions')
|
| 55 |
+
predictions = run_prediction([query], paragraph, 'marshmellow77/roberta-base-cuad',n_best_size=5)
|
| 56 |
answer = ""
|
| 57 |
answer_p=""
|
| 58 |
if predictions['0'] == "":
|