fix: allow ESM2 download on HuggingFace Space

embedder.py

@@ -19,8 +19,9 @@ from config import (
     EMB_CACHE_DIR,
 )
 
-os.environ["TRANSFORMERS_OFFLINE"]  = "1"
+os.environ["TRANSFORMERS_OFFLINE"]  = os.environ.get("FUNGO_OFFLINE", "0")
 os.environ["HF_DATASETS_OFFLINE"]   = "1"
+Path(MODEL_CACHE_DIR).mkdir(parents=True, exist_ok=True)
 os.environ["TRANSFORMERS_CACHE"]    = str(MODEL_CACHE_DIR)
 os.environ["HF_HOME"]               = str(MODEL_CACHE_DIR)
 
@@ -46,13 +47,13 @@ def _load_model():
     _tokenizer = EsmTokenizer.from_pretrained(
         MODEL_NAME,
         cache_dir=MODEL_CACHE_DIR,
-        local_files_only=True,
+        local_files_only=False,
     )
     _model = EsmModel.from_pretrained(
         MODEL_NAME,
         cache_dir=MODEL_CACHE_DIR,
         output_hidden_states=True,
-        local_files_only=True,
+        local_files_only=False,
     )
 
     if USE_FP16: