Update app.py

app.py

@@ -395,7 +395,7 @@ def generation_loop(video_path='./test_videos', seed=666, fps_out=12, batch_size
             sample = sample.to(torch.uint8).numpy()
 
             if is_image:
-                mediapy.write(output_dir, sample[0])
+                mediapy.write_image(output_dir, sample[0])
             else:
                 mediapy.write_video(
                     output_dir, sample, fps=fps_out