Update main.py

main.py

@@ -1,5 +1,11 @@
 import os
 
+# Hugging Face Spaces: redirect cache to /tmp
+os.environ["HF_HUB_CACHE"] = "/tmp/hf_cache"
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/hf_cache"
+
+import os
+
 # Redirect Hugging Face cache to local project directory (not /)
 os.environ["HF_HOME"] = "./hf_cache"
 os.environ["TRANSFORMERS_CACHE"] = "./hf_cache"