fix: utility

app.py

@@ -189,7 +189,7 @@ def generate(
             image = utils.get_image_for_img_to_img(pipeline_to_benchmark)
             call_args.update({"image": image})
         elif "Inpainting" in pipeline_to_benchmark:
-            image, mask_image = utils.get_image_from_inpainting(pipeline_to_benchmark)
+            image, mask_image = utils.get_image_for_inpainting(pipeline_to_benchmark)
             call_args.update({"image": image, "mask_image": mask_image})
         elif "ControlNet" in pipeline_to_benchmark:
             image = utils.get_image_for_controlnet(pipeline_to_benchmark)

utils.py

@@ -12,13 +12,13 @@ def get_image_for_img_to_img(pipeline_to_benchmark):
     return init_image
 
 
-def get_image_from_inpainting(pipeline_to_benchmark):
-    if pipeline_to_benchmark == "SD I2I":
+def get_image_for_inpainting(pipeline_to_benchmark):
+    if pipeline_to_benchmark == "SD Inpainting":
         image_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo.png"
         mask_url = "https://raw.githubusercontent.com/CompVis/latent-diffusion/main/data/inpainting_examples/overture-creations-5sI6fQgYIuo_mask.png"
         init_image = load_image(image_url).convert("RGB").resize((512, 512))
         mask_image = load_image(mask_url).convert("RGB").resize((512, 512))
-    elif pipeline_to_benchmark == "SDXL I2I":
+    elif pipeline_to_benchmark == "SDXL Inpainting":
         image_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/sdxl-text2img.png"
         mask_url = "https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/diffusers/sdxl-inpaint-mask.png"
         init_image = load_image(image_url).convert("RGB")