Update app.py

app.py

@@ -1,9 +1,11 @@
 import os
 
-os.environ["HF_HOME"] = "/tmp/huggingface_cache"
-os.environ["TRANSFORMERS_CACHE"] = "/tmp/huggingface_cache"
-cache_dir = os.environ["HF_HOME"]
+os.environ["HF_HOME"] = "/tmp/huggingface_cache"  # Writable directory in container
 
+# Ensure the cache directory exists
+cache_dir = os.environ["HF_HOME"]
+if not os.path.exists(cache_dir):
+    os.makedirs(cache_dir, exist_ok=True)
 
 if not os.path.exists(cache_dir):
     os.makedirs(cache_dir, exist_ok=True)