Shredder commited on
Commit
f2d5ecf
1 Parent(s): b4079f4

Update Cuad_others.py

Browse files
Files changed (1) hide show
  1. 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, 'Rakib/roberta-base-on-cuad',n_best_size=5)
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'] == "":