Update app.py

app.py

@@ -1,13 +1,14 @@
 import os
 MY_SECRET_TOKEN=os.environ.get('HF_TOKEN_SD')
-
-
+os.system("git clone https://github.com/huggingface/diffusers")
+import sys
+sys.path.append("diffusers")
 import gradio as gr
 import torch
 #from torch import autocast // only for GPU
 from PIL import Image
-from diffusers import StableDiffusionPipeline
-#from diffusers import StableDiffusionImg2ImgPipeline
+#from diffusers import StableDiffusionPipeline
+from diffusers import StableDiffusionImg2ImgPipeline
 
 print("hello sylvain")
 
@@ -15,7 +16,7 @@ YOUR_TOKEN=MY_SECRET_TOKEN
 
 device="cpu"
 
-pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
+pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
 pipe.to(device)
 
 def infer(prompt, init_image):