Update to runway SD 1.5

app.py

@@ -1,4 +1,4 @@
-from diffusers import StableDiffusionInpaintPipeline
+from diffusers import DiffusionPipeline
 import gradio as gr
 import numpy as np
 import imageio
@@ -15,7 +15,7 @@ YOUR_TOKEN=MY_SECRET_TOKEN
 
 device="cpu"
 
-pipe = StableDiffusionInpaintPipeline.from_pretrained("CompVis/stable-diffusion-v1-4", use_auth_token=YOUR_TOKEN)
+pipe = StableDiffusionInpaintPipeline.from_pretrained("runwayml/stable-diffusion-inpainting", use_auth_token=YOUR_TOKEN)
 pipe.to(device)
 
 source_img = gr.Image(source="upload", type="numpy", tool="sketch", elem_id="source_container");
@@ -41,7 +41,7 @@ def predict(source_img, prompt):
     images_list = pipe([prompt] * 2, init_image=src, mask_image=mask, strength=0.75)
     images = []
     safe_image = Image.open(r"unsafe.png")
-    for i, image in enumerate(images_list["sample"]):
+    for i, image in enumerate(images_list["images"]):
         if(images_list["nsfw_content_detected"][i]):
             images.append(safe_image)
         else: