Update app.py

app.py

@@ -4,11 +4,11 @@ from huggingface_hub import hf_hub_download
 import os
 
     
-def extract_spannet_scores(path,input_sentence,length, pos_col, task_col):
+def extract_spannet_scores(model,input_sentence,length, pos_col, task_col):
 
     all_scores = []
     
-    model = SpanNet.load_model(model_path=path)
+    # model = SpanNet.load_model(model_path=path)
     scores = []
     model.eval()
         
@@ -28,10 +28,10 @@ def pool_span_scores(score_dicts, sent_lens):
     return r 
 
 def predict_label(text, model):
-    model_path = 'models/span.model'
+    # model_path = 'models/span.model'
     ip = text.split()
     ip_len = [len(ip)]
-    scores = extract_spannet_scores(model_path,ip,ip_len, pos_col=1, task_col=2)
+    scores = extract_spannet_scores(model,ip,ip_len, pos_col=1, task_col=2)
     pooled_scores = pool_span_scores(scores, ip_len)
     output=''
     for op in pooled_scores[0]:
@@ -54,10 +54,10 @@ if __name__ == '__main__':
                     token=space_key)
 
     from src.network import SpanNet
+    
     model_path = 'models/span.model'
     model = SpanNet.load_model(model_path)
-    # iface = gr.Interface(fn=temp, inputs="text", outputs="text", batch=False)
-    iface = gr.Interface(fn=predict_label(model=model), inputs="text", outputs="text")
-    iface.launch(share=True, blocked_paths=['models'])
+    
+    iface = gr.Interface(fn=predict_label, inputs=["text", model], outputs="text")
     iface.launch(show_api=False)