Update app.py

app.py

@@ -7,6 +7,12 @@ import io
 import base64
 from typing import Optional
 import uvicorn
+import os
+
+# Set cache directories to writable locations
+os.environ["TRANSFORMERS_CACHE"] = "/tmp/transformers_cache"
+os.environ["HF_HOME"] = "/tmp/hf_home"
+os.environ["HF_DATASETS_CACHE"] = "/tmp/hf_datasets_cache"
 
 # Initialize FastAPI app
 app = FastAPI(title="Stable Diffusion API")
@@ -26,7 +32,11 @@ model_id = "CompVis/stable-diffusion-v1-4"
 
 # Check if CUDA is available
 device = "cuda" if torch.cuda.is_available() else "cpu"
-pipe = StableDiffusionPipeline.from_pretrained(model_id, use_auth_token=True)
+pipe = StableDiffusionPipeline.from_pretrained(
+    model_id, 
+    cache_dir="/tmp/diffusers_cache",
+    token=os.environ.get("HF_TOKEN")  # Use token from environment variable
+)
 pipe = pipe.to(device)
 
 @app.get("/")