Update src/streamlit_app.py

src/streamlit_app.py

@@ -1,3 +1,14 @@
+import os
+
+# 1. Create the folder if it doesn't exist
+cache_root = "/mnt/data/huggingface"
+os.makedirs(cache_root, exist_ok=True)
+
+# 2. Redirect Hugging Face Hub, Transformers & Datasets caches
+os.environ["HF_HOME"]            = cache_root
+os.environ["TRANSFORMERS_CACHE"] = os.path.join(cache_root, "transformers")
+os.environ["HF_DATASETS_CACHE"]  = os.path.join(cache_root, "datasets")
+
 import streamlit as st
 # Import the high-level pipeline API from Hugging Face Transformers
 # It simplifies loading models/tokenizers and running common tasks