nehalelkaref commited on
Commit
98173fb
·
1 Parent(s): ed540bf

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,7 +14,7 @@ def extract_spannet_scores(path,input_sentence,length, pos_col, task_col):
14
  scores = []
15
  model.eval()
16
 
17
- out_dict = model(sentences=[sent], output_span_scores=True)
18
  scores.extend([[t.tolist() for t in o[:l]] for o, l in zip(out_dict['span_scores'], length)])
19
  all_scores.append(scores)
20
  return all_scores
 
14
  scores = []
15
  model.eval()
16
 
17
+ out_dict = model(sentences=[input_sentence], output_span_scores=True)
18
  scores.extend([[t.tolist() for t in o[:l]] for o, l in zip(out_dict['span_scores'], length)])
19
  all_scores.append(scores)
20
  return all_scores