Update app.py

app.py

@@ -10,24 +10,22 @@ from diffusers import StableDiffusionPipeline
 import torch
 from dotenv import load_dotenv
 
-# Set local cache to avoid permission issues
-os.environ["HF_HOME"] = os.path.join(os.getcwd(), "hf_cache")
-os.environ["TRANSFORMERS_CACHE"] = os.environ["HF_HOME"]
-os.environ["DIFFUSERS_CACHE"] = os.environ["HF_HOME"]
+# ✅ Set writable cache directory to avoid permission issues
+os.environ["HF_HOME"] = "/tmp/hf_cache"
 
 app = Flask(__name__)
 
-# Load environment variables
+# ✅ Load environment variables
 load_dotenv()
 
-# Configure Gemini
+# ✅ Configure Gemini API
 api_key = os.getenv("GEMINI_API_KEY")
 if api_key:
     genai.configure(api_key=api_key)
 else:
     print("⚠️ GEMINI_API_KEY not found in environment variables!")
 
-# Initialize Stable Diffusion pipeline
+# ✅ Initialize Stable Diffusion pipeline
 pipe = None
 try:
     pipe = StableDiffusionPipeline.from_pretrained(