Update app.py

app.py

@@ -51,6 +51,7 @@ import math
 # from PIL import Image
 import hashlib
 import gradio as gr
+import moviepy.editor as mp
 
 ## load clip model and datasets
 is_cuda = torch.cuda.is_available()
@@ -222,8 +223,10 @@ def render(motions, device_id=0, name='test_vis'):
 
     out = np.stack(vid, axis=0)
     imageio.mimwrite(f'output/results.gif', out, fps=20)
+    clip = mp.VideoFileClip(f'output/results.gif')
+    clip.write_videofile("output/results.mp4")
     del out, vertices
-    return f'output/results.gif'
+    return f'output/results.mp4'
 
 def predict(clip_text, method='fast'):
     gc.collect()
@@ -239,7 +242,9 @@ def predict(clip_text, method='fast'):
     if method == 'fast':
         xyz = pred_xyz.reshape(1, -1, 22, 3)
         pose_vis = plot_3d.draw_to_batch(xyz.detach().cpu().numpy(), title_batch=None, outname=[f'output/results.gif'])
-        return f'output/results.gif'
+        clip = mp.VideoFileClip("output/results.gif")
+        clip.write_videofile("output/results.mp4")
+        return f'output/results.mp4'
     elif method == 'slow':
         output_path = render(pred_xyz.detach().cpu().numpy().squeeze(axis=0), device_id=0, name=output_name)
         return output_path