Update app.py

app.py

@@ -15,20 +15,21 @@ def resize(value,img):
     img = img.resize((value,value))
     return img
 
-def predict(source_img, prompt):
+def predict(source_img, prompt, strength):
     imageio.imwrite("data.png", source_img['image'])
     imageio.imwrite("data_mask.png", source_img["mask"])
     src = resize(768, "data.png")
     src.save("src.png")
     mask = resize(768, "data_mask.png")  
     mask.save("mask.png")
-    image = pipe(prompt=prompt, image=src, mask_image=mask, num_inference_steps=5, strength=0.8, guidance_scale=0.0).images[0]
+    image = pipe(prompt=prompt, image=src, mask_image=mask, num_inference_steps=5, strength=strength, guidance_scale=0.0).images[0]
     return 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=[gr.Image(source=("upload"), tool='sketch', label='Source Image'), 
-                                 gr.Textbox(label='What you want the AI to Generate, 77 Token limit')], 
+                                gr.Textbox(label='What you want the AI to Generate, 77 Token limit'),
+                                gr.Slider(minimum=.5, maximum=1, value=.5, step=.25, label='Strength')], 
              outputs='image', 
              title=title, 
              description=description,