kleinhe commited on
Commit
7bdbc03
·
1 Parent(s): 479b355
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -73,7 +73,10 @@ def demo(prompt, mode, condition, render_mode="joints", skip_steps=0, out_size=1
73
  meshes, _ = visual.get_mesh(output)
74
  pics = visual.pyrender_process(meshes, out_size, out_size)
75
 
76
- imageio.mimsave(video_path, pics, duration= 1000 / 20, loop=0)
 
 
 
77
 
78
  if mode[i] == "cadm":
79
  out_paths[0] = video_path
 
73
  meshes, _ = visual.get_mesh(output)
74
  pics = visual.pyrender_process(meshes, out_size, out_size)
75
 
76
+ try:
77
+ imageio.mimsave(video_path, pics, duration= 1000 / 20, loop=0)
78
+ except:
79
+ imageio.mimsave(video_path, pics, fps=20, loop=0)
80
 
81
  if mode[i] == "cadm":
82
  out_paths[0] = video_path