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):
+def predict_label(text, model=model):
     model_path = 'models/span.model'
     ip = text.split()
     ip_len = [len(ip)]
@@ -42,8 +42,8 @@ def temp(text):
     print('IN FUNCTION')
     return text
 print('STARTING ..')
-# model_path = 'models/span.model'
-# model = SpanNet.load_model(model_path)
+
+
 if __name__ == '__main__':
     space_key = os.environ.get('key')
     filenames = ['network.py', 'layers.py', 'utils.py', 'representation.py']
@@ -54,8 +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, inputs="text", outputs="text",auth=True)
+    iface = gr.Interface(fn=predict_label, inputs="text", outputs="text")
     iface.launch(share=True, blocked_paths=['models'])
     iface.launch(show_api=False)