vumichien commited on
Commit
5457346
1 Parent(s): 601e723

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -223,8 +223,8 @@ def render(motions, device_id=0, name='test_vis'):
223
 
224
  out = np.stack(vid, axis=0)
225
  imageio.mimwrite(f'output/results.gif', out, fps=20)
226
- clip = mp.VideoFileClip(f'output/results.gif')
227
- clip.write_videofile("output/results.mp4")
228
  del out, vertices
229
  return f'output/results.mp4'
230
 
@@ -242,8 +242,8 @@ def predict(clip_text, method='fast'):
242
  if method == 'fast':
243
  xyz = pred_xyz.reshape(1, -1, 22, 3)
244
  pose_vis = plot_3d.draw_to_batch(xyz.detach().cpu().numpy(), title_batch=None, outname=[f'output/results.gif'])
245
- clip = mp.VideoFileClip("output/results.gif")
246
- clip.write_videofile("output/results.mp4")
247
  return f'output/results.mp4'
248
  elif method == 'slow':
249
  output_path = render(pred_xyz.detach().cpu().numpy().squeeze(axis=0), device_id=0, name=output_name)
 
223
 
224
  out = np.stack(vid, axis=0)
225
  imageio.mimwrite(f'output/results.gif', out, fps=20)
226
+ out_video = mp.VideoFileClip(f'output/results.gif')
227
+ out_video.write_videofile("output/results.mp4")
228
  del out, vertices
229
  return f'output/results.mp4'
230
 
 
242
  if method == 'fast':
243
  xyz = pred_xyz.reshape(1, -1, 22, 3)
244
  pose_vis = plot_3d.draw_to_batch(xyz.detach().cpu().numpy(), title_batch=None, outname=[f'output/results.gif'])
245
+ out_video = mp.VideoFileClip("output/results.gif")
246
+ out_video.write_videofile("output/results.mp4")
247
  return f'output/results.mp4'
248
  elif method == 'slow':
249
  output_path = render(pred_xyz.detach().cpu().numpy().squeeze(axis=0), device_id=0, name=output_name)