Update app.py

app.py

@@ -3,7 +3,7 @@ from PIL import Image
 from io import BytesIO
 import torch
 import os
-import imagic
+
 from diffusers import DiffusionPipeline, DDIMScheduler, UNet2DConditionModel
 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
 
@@ -13,7 +13,7 @@ device = "cuda"
 pipe = DiffusionPipeline.from_pretrained(
     "CompVis/stable-diffusion-v1-4",
     safety_checker=None,
-    custom_pipeline="https://huggingface.co/spaces/fffiloni/imagic-stable-diffusion/blob/main/imagic.py",
+    custom_pipeline="imagic_sylvain",
     scheduler = DDIMScheduler(beta_start=0.00085, beta_end=0.012, beta_schedule="scaled_linear", clip_sample=False, set_alpha_to_one=False)
 ).to(device)