Update app.py

app.py

@@ -44,19 +44,18 @@ def temp(text):
 print('STARTING ..')
 # model_path = 'models/span.model'
 # model = SpanNet.load_model(model_path)
-space_key = os.environ.get('key')
-hf_hub_download(repo_id="nehalelkaref/stagedNER", token=space_key,filename='src/network.py', local_dir='src')
-hf_hub_download(repo_id="nehalelkaref/stagedNER", token=space_key,filename='src/utils.py', local_dir='src2')
-
-# hf_hub_download(repo_id="nehalelkaref/stagedNER", token=space_key,filename='src/utils.py')
-
-from src.network import SpanNet
-from src2.utils import *
-
-# gr.load(name="nehalelkaref/flat-arabic-entity-classification", hf_token=space_key, src='spaces')
-iface = gr.Interface(fn=temp, inputs="text", outputs="text", batch=False)
-# iface = gr.Interface(fn=predict_label, inputs="text", outputs="text",auth=True)
-iface.launch(share=True, blocked_paths=['models'])
-
-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
+    # iface = gr.Interface(fn=temp, inputs="text", outputs="text", batch=False)
+    iface = gr.Interface(fn=predict_label, inputs="text", outputs="text",auth=True)
+    iface.launch(share=True, blocked_paths=['models'])
+    iface.launch(show_api=False)