Fix looping

utils.py

@@ -85,12 +85,8 @@ def create_gif(frames, fps, rescale=False, path=None, watermark=None):
         outputs.append(x)
         # imageio.imsave(os.path.join(dir, os.path.splitext(name)[0] + f'_{i}.jpg'), x)
 
-    imageio.mimsave(path, outputs, duration=1/fps)
+    imageio.mimsave(path, outputs, loop=0, duration=1000/fps)
     return path
 
 
-def post_process_gif(list_of_results, image_resolution):
-    output_file = "/tmp/ddxk.gif"
-    imageio.mimsave(output_file, list_of_results, duration=1/4)
-    return output_file