Update app.py

app.py

@@ -5,7 +5,7 @@ from torch import autocast
 from diffusers import StableDiffusionPipeline, DDIMScheduler
 from IPython.display import display
 
-model_path = WEIGHTS_DIR             # If you want to use previously trained model saved in gdrive, replace this with the full path of model in gdrive
+model_path = "slymnyldrm/dreambooth_usecase_weights"
 
 pipe = StableDiffusionPipeline.from_pretrained(model_path, safety_checker=None, torch_dtype=torch.float16).to("cuda")
 pipe.scheduler = DDIMScheduler.from_config(pipe.scheduler.config)