Update app.py

app.py

@@ -12,10 +12,10 @@ from imagic import ImagicStableDiffusionPipeline
 has_cuda = torch.cuda.is_available()
 device = "cuda" 
 
-pipe = DiffusionPipeline.from_pretrained(
+pipe = ImagicStableDiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
     safety_checker=None,
-    custom_pipeline=ImagicStableDiffusionPipeline,
+    #custom_pipeline=ImagicStableDiffusionPipeline,
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)