Enable batch size

app.py

@@ -20,8 +20,8 @@ SYSTEM_PROMPT = r"""This two-panel split-frame image showcases a furniture in as
 
 if not torch.cuda.is_available():
 
-    def _dummy_pipe(image: Image.Image, *args, **kwargs):  # noqa: ARG001
-        return {"images": [image]}
+    def _dummy_pipe(image: list[Image.Image], *args, **kwargs):  # noqa: ARG001
+        return {"images": image}
 
     pipe = _dummy_pipe
 else:
@@ -59,7 +59,7 @@ def calculate_optimal_dimensions(image: Image.Image) -> tuple[int, int]:
 def infer(
     furniture_image: Image.Image,
     room_image: EditorValue,
-    prompt: str,
+    prompt: str = "",
     seed: int = 42,
     randomize_seed: bool = False,
     guidance_scale: float = 3.5,
@@ -132,15 +132,16 @@ def infer(
     if randomize_seed:
         seed = secrets.randbelow(MAX_SEED)
 
+    prompt = prompt + ".\n" + SYSTEM_PROMPT if prompt else SYSTEM_PROMPT
+    batch_size = 4
     results_images = pipe(
-        prompt=prompt + ".\n" + SYSTEM_PROMPT,
-        image=image,
-        mask_image=mask,
+        prompt=[prompt] * batch_size,
+        image=[image] * batch_size,
+        mask_image=[mask] * batch_size,
         height=height,
         width=width,
         guidance_scale=guidance_scale,
         num_inference_steps=num_inference_steps,
-        batch_size=4,
         generator=torch.Generator("cpu").manual_seed(seed),
     )["images"]
 

pyproject.toml

@@ -18,4 +18,4 @@ dependencies = [
 ]
 
 [tool.uv.sources]
-diffusers = { git = "https://github.com/huggingface/diffusers.git" }
+diffusers = { git = "https://github.com/huggingface/diffusers.git" }