Spaces:
Build error
Build error
Commit
·
5c5fdba
1
Parent(s):
795587f
Update app.py
Browse files
app.py
CHANGED
@@ -7,9 +7,6 @@ from huggingface_hub import Repository
|
|
7 |
|
8 |
def extract_spannet_scores(path,input_sentence,length, pos_col, task_col):
|
9 |
|
10 |
-
# sent = input_sentence.split()
|
11 |
-
# length = [len(sent)]
|
12 |
-
|
13 |
all_scores = []
|
14 |
|
15 |
model = SpanNet.load_model(model_path=path)
|
@@ -37,7 +34,10 @@ def predict_label(text):
|
|
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 |
-
|
|
|
|
|
|
|
41 |
|
42 |
iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
|
43 |
iface.launch()
|
|
|
7 |
|
8 |
def extract_spannet_scores(path,input_sentence,length, pos_col, task_col):
|
9 |
|
|
|
|
|
|
|
10 |
all_scores = []
|
11 |
|
12 |
model = SpanNet.load_model(model_path=path)
|
|
|
34 |
ip_len = [len(ip)]
|
35 |
scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
|
36 |
pooled_scores = pool_span_scores(scores, ip_len)
|
37 |
+
output=''
|
38 |
+
for op in pooled_scores[0]:
|
39 |
+
output+= op + ','
|
40 |
+
return output
|
41 |
|
42 |
iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
|
43 |
iface.launch()
|