Update app.py

app.py

@@ -1,3 +1,27 @@
+
+Hugging Face's logo Hugging Face
+Spaces:
+Manjushri
+/
+SDXL-Turbo-Inpainting-CPU
+App
+Files
+Community
+3
+SDXL-Turbo-Inpainting-CPU
+/ app.py
+Manjushri's picture
+Manjushri
+Update app.py
+4e4f2f6
+verified
+raw
+history
+blame
+contribute
+delete
+No virus
+1.59 kB
 from diffusers import StableDiffusionXLInpaintPipeline
 import gradio as gr
 import numpy as np
@@ -20,13 +44,13 @@ def img2img(source_img, prompt, strength):
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
     src.save("src.png")
-    mask = resize(768, "data_mask.png")
+    mask = resize(768, "data_mask.png")  
     mask.save("mask.png")
     image = pipe(prompt=prompt, image=src, mask_image=mask, num_inference_steps=6, strength=strength, guidance_scale=0.0).images[0]
     return image
-
-title="GoldExtra Zeichnung zu 2.5D"
-description="OpenCV.js > i2i > 3D"
+    
+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=img2img, inputs=[gr.Image(source="upload", tool="sketch", label="Source Image"), 
                                 gr.Textbox(label='What you want the AI to Generate, 77 Token limit'),
                                 gr.Slider(minimum=.5, maximum=1, value=.75, step=.025, label='Strength')],