Update app.py

app.py

@@ -236,12 +236,13 @@ class Model:
     def gen_video(self, w1, w2, noise, path, frame_num=10):
         video = imageio.get_writer(path, mode='I', fps=frame_num // 2, codec='libx264', bitrate='16M')
         lin = np.linspace(0, 1, frame_num)
+        frames = []
         for i in range(0, frame_num):
             img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
+            frames.append(img)
             video.append_data(img)
         for i in reversed(range(0, frame_num)):
-            img = self.get_img(((1 - lin[i]) * w1) + (lin[i] * w2), noise)
-            video.append_data(img)
+            video.append_data(frames[i])
         video.close()
 
 def get_thumbnail(img):