Update app.py

app.py

@@ -16,10 +16,10 @@ YOUR_TOKEN=MY_SECRET_TOKEN
 
 device="cpu"
 
-#prompt_pipe = StableDiffusionPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
+#prompt_pipe = StableDiffusionPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=YOUR_TOKEN)
 #prompt_pipe.to(device)
 
-img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
+img_pipe = StableDiffusionImg2ImgPipeline.from_pretrained("runwayml/stable-diffusion-v1-5", use_auth_token=YOUR_TOKEN)
 img_pipe.to(device)
 
 source_img = gr.Image(source="upload", type="filepath", label="init_img | 512*512 px")