Spaces:
Build error
Build error
Commit
·
59a84ff
1
Parent(s):
3d70b45
Update app.py
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ def pool_span_scores(score_dicts, sent_lens):
|
|
34 |
def predict_label(text):
|
35 |
model_path = 'models/span_model'
|
36 |
ip = text.split()
|
37 |
-
ip_len = [
|
38 |
scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
|
39 |
pooled_scores = pool_span_scores(scores, ip_len)
|
40 |
return pooled_scores
|
|
|
34 |
def predict_label(text):
|
35 |
model_path = 'models/span_model'
|
36 |
ip = text.split()
|
37 |
+
ip_len = [len(ip)]
|
38 |
scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
|
39 |
pooled_scores = pool_span_scores(scores, ip_len)
|
40 |
return pooled_scores
|