Update app.py

app.py

@@ -1,10 +1,16 @@
 """Demo for NER4OPT."""
+import os
 import warnings
 
 warnings.filterwarnings("ignore", category=DeprecationWarning)
 
 from huggingface_hub import snapshot_download
 
+# Set custom cache directory BEFORE importing anything
+custom_cache = "./my_hf_cache"
+os.environ["HF_HOME"] = custom_cache
+os.environ["TRANSFORMERS_CACHE"] = custom_cache
+
 # 1. Force Download: Due to issues from huggingface spaces
 try:
     model_name_one = "skadio/ner4opt-roberta-v1"
@@ -20,7 +26,6 @@ try:
 except Exception as e:
     print(f"Download failed: {e}")
 
-import os
 # Step 2: Set environment variables to force offline mode GLOBALLY
 os.environ["HF_HUB_OFFLINE"] = "1"
 os.environ["TRANSFORMERS_OFFLINE"] = "1"