Update app.py

app.py

@@ -63,20 +63,17 @@ def interpolate(save_dir='./lerp/', frames=100, rows=8, cols=8):
 
     zs += zs[::-1]  # also go in reverse order to complete loop
 
+    frames = []
     for i, z in enumerate(zs):
         imgs = model(z)
 
-        # normalize
-        imgs = (imgs + 1) / 2
-
-        imgs = (imgs.permute(0, 2, 3, 1).cpu().numpy() * 255).astype(np.uint8)
-
-        # create grid
-        imgs = einops.rearrange(imgs, "(b1 b2) h w c -> (b1 h) (b2 w) c", b1=rows, b2=cols)
-
-        Image.fromarray(imgs).save(save_dir / f"{i:03}.png")
-
-    subprocess.call(f"convert -dispose previous -delay 10 -loop 0 {save_dir}/*.png out.gif".split())
+        save_image(imgs, save_dir / f"{i:03}.png", normalize=True)
+        img = Image.open(save_dir / f"{i:03}.png").convert('RGBA')
+        img.putalpha(255)
+        frames.append(img)
+        img.save(save_dir / f"{i:03}.png")
+    frames[0].save("out.gif", format="GIF", append_images=frames,
+                   save_all=True, duration=100, loop=1)
 
 
 def predict(choice, seed):