Update app.py

app.py

@@ -3,24 +3,33 @@ import warnings
 
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 
-import streamlit as st
-import ner4opt
-from ner4opt.utils import preprocess, spacy_tokenize_sentence
 from huggingface_hub import snapshot_download
-from spacy import displacy
-
-HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
 
 # Force Download: Due to issues from huggingface spaces
 try:
     model_name_one = "skadio/ner4opt-roberta-v1"
-    snapshot_download(repo_id=model_name_one)
+    snapshot_download(repo_id=model_name_one, 
+                      resume_download=True,  # Resume if interrupted
+                      local_files_only=False  # Allow download if not cached
+                     )
     model_name_two = "skadio/ner4opt-roberta-v2"
-    snapshot_download(repo_id=model_name_two)
+    snapshot_download(repo_id=model_name_two,
+                      resume_download=True,  # Resume if interrupted
+                      local_files_only=False  # Allow download if not cached
+                     )
 except Exception as e:
     print(f"Download failed: {e}")
 
 
+import streamlit as st
+import ner4opt
+from ner4opt.utils import preprocess, spacy_tokenize_sentence
+
+from spacy import displacy
+
+HTML_WRAPPER = """<div style="overflow-x: auto; border: 1px solid #e6e9ef; border-radius: 0.25rem; padding: 1rem; margin-bottom: 2.5rem">{}</div>"""
+
+
 @st.cache_resource
 def load_models():
     """Load and cache NER4OPT models."""