Update app.py

app.py

@@ -16,7 +16,7 @@ def resize(value,img):
     return img
 
 def predict(source_img, prompt):
-    imageio.imwrite("data.png", source_img[simage])
+    imageio.imwrite("data.png", source_img["image"])
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
     src.save("src.png")
@@ -25,10 +25,9 @@ def predict(source_img, prompt):
     image = pipe(prompt=prompt, image=src, mask_image=mask, num_inference_steps=4, strength=0.5, guidance_scale=0.0).images[0]
     return image
 
-simage = gr.ImageEditor(sources=("upload"), brush=gr.Brush(colors=["#000000"]), label='Source Image')
 title="SDXL Turbo Inpainting CPU"
 description="Inpainting with SDXL Turbo <br><br> <b>Please use square .png image as input, 512x512, 768x768, or 1024x1024</b>"
-gr.Interface(fn=predict, inputs=[simage, 
+gr.Interface(fn=predict, inputs=[gr.ImageEditor(sources=("upload"), brush=gr.Brush(colors=["#000000"]), label='Source Image'), 
                                  gr.Textbox(label='What you want the AI to Generate, 77 Token limit')], 
              outputs='image', 
              title=title,