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'] == "":
|