app

app.py

@@ -234,24 +234,22 @@ def infer(
         samples = unpreprocess(image_unprocessed).contiguous()
 
 
-        to_pil = ToPILImage()
-        pil_images = [to_pil(img) for img in samples]
-
-        # Get the first and last images
-        first_image = pil_images[0]
-        last_image = pil_images[-1]
-
-        # Create an animated GIF from all the PIL images
-        # gif_buffer = io.BytesIO()
-        # # duration: adjust duration (ms) between frames as needed
-        # pil_images[0].save(gif_buffer, format="GIF", save_all=True, append_images=pil_images[1:], duration=10, loop=0)
-        # gif_buffer.seek(0)
-        # gif_bytes = gif_buffer.read()
-
-        # return first_image, last_image, gif_bytes, seed
-        return first_image, last_image, seed
-
-    return pil_image, seed
+    to_pil = ToPILImage()
+    pil_images = [to_pil(img) for img in samples]
+
+    # Get the first and last images
+    first_image = pil_images[0]
+    last_image = pil_images[-1]
+
+    # Create an animated GIF from all the PIL images
+    # gif_buffer = io.BytesIO()
+    # # duration: adjust duration (ms) between frames as needed
+    # pil_images[0].save(gif_buffer, format="GIF", save_all=True, append_images=pil_images[1:], duration=10, loop=0)
+    # gif_buffer.seek(0)
+    # gif_bytes = gif_buffer.read()
+
+    # return first_image, last_image, gif_bytes, seed
+    return first_image, last_image, seed
 
 
 # examples = [