sijdsohs

app.py

@@ -13,7 +13,7 @@ DEVICE = "cuda"
 
 MAX_SEED = np.iinfo(np.int32).max
 FIXED_DIMENSION = 900
-# FIXED_DIMENSION = 512 + (512 // 2)
+FIXED_DIMENSION = 720
 FIXED_DIMENSION = (FIXED_DIMENSION // 16) * 16
 
 SYSTEM_PROMPT = r"""This two-panel split-frame image showcases a furniture in as a product shot versus styled in a room.
@@ -22,8 +22,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: list[Image.Image], *args, **kwargs):  # noqa: ARG001
-        return {"images": image}
+    def _dummy_pipe(image: Image.Image, *args, **kwargs):  # noqa: ARG001
+        return {"images": [image]}
 
     pipe = _dummy_pipe
 else:
@@ -48,7 +48,7 @@ else:
     pipe.to(DEVICE)
 
 
-@spaces.GPU
+@spaces.GPU(duration=120)
 def infer(
     furniture_image: Image.Image,
     room_image: EditorValue,