Update app.py

app.py

@@ -38,7 +38,7 @@ def predict(prompt, source_img):
     mask = resize(512, "data_mask.png")  
     mask.save("mask.png")
     
-    images_list = pipe([prompt] * 1, init_image=src, mask_image=mask, strength=0.75)
+    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"]):
@@ -49,5 +49,6 @@ def predict(prompt, source_img):
     return images
 
 custom_css="style.css"
-
-gr.Interface(fn=predict, inputs=["text", source_img], outputs=gallery, css=custom_css).launch(enable_queue=True)
+title="InPainting Stable Diffusion CPU"
+description="Inpainting Stable Diffusion example using CPU and HF token. <br />Warning: Slow process... ~5/10 min inference time. <b>NSFW filter enabled.</b><br /><span style="color:'red';">Please use 512*512 square image as input to avoid memory error </span>" 
+gr.Interface(fn=predict, inputs=["text", source_img], outputs=gallery, css=custom_css, title=title, description=description).launch(enable_queue=True)