Update app.py

app.py

@@ -6,10 +6,20 @@ 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)
+    model_name_two = "skadio/ner4opt-roberta-v2"
+    snapshot_download(repo_id=model_name_two)
+except Exception as e:
+    print(f"Download failed: {e}")
+
 
 @st.cache_resource
 def load_models():