Update app.py

app.py

@@ -27,7 +27,7 @@ def pool_span_scores(score_dicts, sent_lens):
     r = [[TAGS[ps] for ps in sent_ps] for sent_ps in pooled_scores]
     return r 
 
-def predict_label(text, model=model):
+def predict_label(text):
     # model_path = 'models/span.model'
     ip = text.split()
     ip_len = [len(ip)]
@@ -44,19 +44,19 @@ def temp(text):
 print('STARTING ..')
 
 
-
-space_key = os.environ.get('key')
-filenames = ['network.py', 'layers.py', 'utils.py', 'representation.py']
-for file in filenames:
-    hf_hub_download('nehalelkaref/stagedNER',
-                filename=file,
-                local_dir='src',
-                token=space_key)
-
-from src.network import SpanNet
-
-model_path = 'models/span.model'
-model = SpanNet.load_model(model_path)
-
-iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
-iface.launch(show_api=False)
+if __name__ == '__main__':
+    space_key = os.environ.get('key')
+    filenames = ['network.py', 'layers.py', 'utils.py', 'representation.py']
+    for file in filenames:
+        hf_hub_download('nehalelkaref/stagedNER',
+                    filename=file,
+                    local_dir='src',
+                    token=space_key)
+    
+    from src.network import SpanNet
+    
+    model_path = 'models/span.model'
+    model = SpanNet.load_model(model_path)
+    
+    iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
+    iface.launch(show_api=False)