Update app.py

app.py

@@ -7,10 +7,10 @@ from diffusers import StableDiffusionPipeline, DDIMScheduler
 import torch
 
 MODEL_NAME = "runwayml/stable-diffusion-v1-5"
-OUTPUT_DIR = "/content/stable_diffusion_weights/custom_model"
+OUTPUT_DIR = "/home/user/app/stable_diffusion_weights/custom_model"
 
 def fine_tune(instance_prompt, image1, image2):
-    instance_data_dir = "/content/instance_images"
+    instance_data_dir = "/home/user/app/instance_images"
     if os.path.exists(instance_data_dir):
         shutil.rmtree(instance_data_dir)
     os.makedirs(instance_data_dir, exist_ok=True)