Update src/streamlit_app.py

src/streamlit_app.py

@@ -5,14 +5,14 @@ import torch.nn.functional as F
 from transformers import AutoTokenizer, AutoModelForCausalLM
 import os
 
-# Use a local writable cache directory inside the repo
-CACHE_DIR = "./.cache_hf"
-os.environ["HF_HOME"] = CACHE_DIR
-os.environ["TRANSFORMERS_CACHE"] = os.path.join(CACHE_DIR, "transformers")
-os.environ["HF_HUB_CACHE"] = os.path.join(CACHE_DIR, "hub")
-
-os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
-os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
+# # Use a local writable cache directory inside the repo
+# CACHE_DIR = "./.cache_hf"
+# os.environ["HF_HOME"] = CACHE_DIR
+# os.environ["TRANSFORMERS_CACHE"] = os.path.join(CACHE_DIR, "transformers")
+# os.environ["HF_HUB_CACHE"] = os.path.join(CACHE_DIR, "hub")
+
+# os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
+# os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
 
 
 HF_MODEL_ID = "ma7583/pvs_oracle"