Spaces:
Build error
Build error
Commit
·
98173fb
1
Parent(s):
ed540bf
Update app.py
Browse files
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=[
|
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
|