Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -194,9 +194,9 @@ def video_inference(
194
  return im
195
 
196
  anim = animation.FuncAnimation(fig, animate, init_func=init, frames=video.shape[0], interval=30)
197
- anim.save("avatar.gif", dpi=300, writer = animation.PillowWriter(fps=24))
198
 
199
- return "avatar.gif"
200
 
201
  with gr.Blocks() as demo:
202
  gr.Markdown("# **<p align='center'>ROME: Realistic one-shot mesh-based head avatars</p>**")
 
194
  return im
195
 
196
  anim = animation.FuncAnimation(fig, animate, init_func=init, frames=video.shape[0], interval=30)
197
+ anim.save("avatar.mp4", dpi=300, writer = animation.PillowWriter(fps=24))
198
 
199
+ return "avatar.mp4"
200
 
201
  with gr.Blocks() as demo:
202
  gr.Markdown("# **<p align='center'>ROME: Realistic one-shot mesh-based head avatars</p>**")