Qwen-Image-Layered

Paused

lainlives commited on May 23

Commit

3648cda

verified ·

1 Parent(s): c0517a8

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -73,7 +73,7 @@ def infer(input_files,
             with torch.inference_mode():
                 output = pipeline(**inputs)
-                output_images = output.images
             # Zip layers for this specific frame
             frame_zip_path = tempfile.NamedTemporaryFile(suffix=".zip", delete=False).name

             with torch.inference_mode():
                 output = pipeline(**inputs)
+                output_images = output.images[0]
             # Zip layers for this specific frame
             frame_zip_path = tempfile.NamedTemporaryFile(suffix=".zip", delete=False).name