Update app.py

app.py

@@ -9,8 +9,8 @@ import torch
 from diffusers import DiffusionPipeline
 from PIL import Image
 
-# Create permanent storage directory
-SAVE_DIR = "/content/generated_images"
+# Create permanent storage directory in the current working directory
+SAVE_DIR = os.path.join(os.getcwd(), "generated_images")
 if not os.path.exists(SAVE_DIR):
     os.makedirs(SAVE_DIR, exist_ok=True)