Update app.py

app.py

@@ -5,10 +5,10 @@ from huggingface_hub import Repository
 def greet(name):
     return "Hello " + name + "!!"
     
-def extract_spannet_scores(path,input_sentence, pos_col, task_col):
+def extract_spannet_scores(path,input_sentence,length, pos_col, task_col):
 
-    sent = input_sentence.split()
-    length = [len(sent)]
+    # sent = input_sentence.split()
+    # length = [len(sent)]
    
     all_scores = []
     
@@ -31,10 +31,13 @@ def pool_span_scores(score_dicts, sent_lens):
     r = [[TAGS[ps] for ps in sent_ps]for sent_ps in pooled_scores]
     return r 
 
-input_text = gr.textbox(label="Text 1")
+def predict_label(text):
+    model_path = 'models/span_model'
+    ip = input_text.split()
+    ip_len = [length(ip)]
+    scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
+    pooled_scores = pool_span_scores(scores, ip_len)
+    return pooled_scores
+
+iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
 
-iface = gr.Interface(fn=greet, inputs=input_text, outputs="text")
-iface.
-model_path = 'models/span_model'
-scores = extract_spannet_scores(model_path,input_text, pos_col=1, task_col=2)
-iface.launch()