Update app.py

app.py

@@ -8,9 +8,12 @@ import re
 import os
 
 
-# FIX CACHE ISSUE ON HUGGINGFACE SPACE
+# Set custom cache directory with proper permissions
 os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
 os.makedirs("/tmp/transformers_cache", exist_ok=True)
+# Also set HF_HOME to avoid permission issues
+os.environ["HF_HOME"] = "/tmp/hf_home"
+os.makedirs("/tmp/hf_home", exist_ok=True)
 
 app = FastAPI()