Upload folder using huggingface_hub

api/main.py

@@ -22,7 +22,8 @@ logger = logging.getLogger(__name__)
 sys.path.insert(0, os.path.dirname(os.path.dirname(os.path.abspath(__file__))))
 
 def download_models():
-    
+    from src.ner import load_ner_model
+    load_ner_model()
     hf_token = os.getenv("HF_TOKEN")
     if not hf_token:
         logger.warning("HF_TOKEN not set — skipping model download.")
@@ -56,9 +57,6 @@ def download_models():
 
 download_models()
 
-from src.ner import load_ner_model
-load_ner_model()
-
 AGENT_VERSION = os.getenv("AGENT_VERSION", "v2")
 
 if AGENT_VERSION == "v2":