cbensimon HF staff commited on
Commit
6d06601
1 Parent(s): 4ebe638

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -12,6 +12,7 @@ import argparse
12
  import imageio
13
  import numpy as np
14
  import gradio as gr
 
15
  import os
16
  from PIL import Image
17
  from subprocess import PIPE, run
@@ -32,6 +33,7 @@ if(is_spaces):
32
 
33
  animator = MagicAnimate()
34
 
 
35
  def animate(reference_image, motion_sequence_state, seed=1, steps=25, guidance_scale=7.5):
36
  return animator(reference_image, motion_sequence_state, seed, steps, guidance_scale)
37
 
 
12
  import imageio
13
  import numpy as np
14
  import gradio as gr
15
+ import spaces
16
  import os
17
  from PIL import Image
18
  from subprocess import PIPE, run
 
33
 
34
  animator = MagicAnimate()
35
 
36
+ @spaces.GPU
37
  def animate(reference_image, motion_sequence_state, seed=1, steps=25, guidance_scale=7.5):
38
  return animator(reference_image, motion_sequence_state, seed, steps, guidance_scale)
39