nehalelkaref commited on
Commit
5c5fdba
·
1 Parent(s): 795587f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
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
- return pooled_scores
 
 
 
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()