Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,5 +1,7 @@
 import os
 
+import os
+
 # Use /tmp (always writable in Hugging Face Spaces & Docker)
 os.environ["HF_HOME"] = "/tmp/huggingface"
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers"
@@ -12,14 +14,14 @@ os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
 
 
 # Redirect Hugging Face cache to writable directory
-os.environ["HF_HOME"] = "/app/.cache/huggingface"
-os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/transformers"
-os.environ["HF_HUB_CACHE"] = "/app/.cache/hub"
+# os.environ["HF_HOME"] = "/app/.cache/huggingface"
+# os.environ["TRANSFORMERS_CACHE"] = "/app/.cache/transformers"
+# os.environ["HF_HUB_CACHE"] = "/app/.cache/hub"
 
 # Make sure the folders exist
-os.makedirs(os.environ["HF_HOME"], exist_ok=True)
-os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
-os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
+# os.makedirs(os.environ["HF_HOME"], exist_ok=True)
+# os.makedirs(os.environ["TRANSFORMERS_CACHE"], exist_ok=True)
+# os.makedirs(os.environ["HF_HUB_CACHE"], exist_ok=True)
 
 import streamlit as st
 import json