fix(api): match panorama img2img parameter check to numpy code path

ssube · Apr 28, 2023 · c8e1519 · c8e1519
1 parent 1d9de2c
commit c8e1519
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/api/onnx_web/diffusers/pipelines/panorama.py b/api/onnx_web/diffusers/pipelines/panorama.py
@@ -848,8 +848,8 @@ def __call__(
         *args,
         **kwargs,
     ):
-        if len(args) > 0 and (
-            isinstance(args[0], np.ndarray) or isinstance(args[0], PIL.Image.Image)
+        if len(args) > 1 and (
+            isinstance(args[1], np.ndarray) or isinstance(args[1], PIL.Image.Image)
         ):
             logger.debug("running img2img panorama pipeline")
             return self.img2img(*args, **kwargs)