Update app.py

app.py

@@ -15,10 +15,15 @@ logging.basicConfig(
 logger = logging.getLogger(__name__)
 
 # Set Hugging Face cache directory to a writable path
-os.environ['HF_HOME'] = '/tmp/.cache/huggingface'
+# Make sure to set this BEFORE importing or initializing any models
+os.environ['TRANSFORMERS_CACHE'] = '/tmp/transformers_cache'
+os.environ['HF_HOME'] = '/tmp/hf_home'
+os.environ['XDG_CACHE_HOME'] = '/tmp/cache'
 
-# Create cache directory if it doesn't exist
-os.makedirs('/tmp/.cache/huggingface', exist_ok=True)
+# Create cache directories if they don't exist
+os.makedirs('/tmp/transformers_cache', exist_ok=True)
+os.makedirs('/tmp/hf_home', exist_ok=True)
+os.makedirs('/tmp/cache', exist_ok=True)
 
 # Global variable for the model
 pipe = None
@@ -30,7 +35,8 @@ def load_model():
         logger.info("Loading Zero123Plus model...")
         pipe = DiffusionPipeline.from_pretrained(
             "sudo-ai/zero123plus-v1.2",
-            torch_dtype=torch.float32
+            torch_dtype=torch.float32,
+            cache_dir="/tmp/diffusers_cache"
         )
         pipe.to("cpu")
         logger.info("Model loaded successfully")