Update app.py

app.py

@@ -1,7 +1,7 @@
 import gradio as gr
 import numpy as np
-from flat-arabic-entity-classification.network import SpanNet
-from huggingface_hub import Repository
+
+from huggingface_hub import hf_hub_download
 import os
 
     
@@ -46,7 +46,10 @@ print('STARTING ..')
 # model_path = 'models/span.model'
 # model = SpanNet.load_model(model_path)
 space_key = os.environ.get('key')
-gr.load(name="nehalelkaref/flat-arabic-entity-classification", hf_token=space_key, src='spaces')
+hf_hub_download(repo_id="nehalelkaref/stagedNER", token=space_key)
+from network import SpanNet
+
+# 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'])