roll-ai commited on
Commit
48e825b
·
verified ·
1 Parent(s): b67ab4c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -61,6 +61,7 @@ print_ckpt_structure()
61
 
62
  from inference.flovd_demo import generate_video
63
 
 
64
  def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pose_name):
65
  log_buffer = io.StringIO()
66
  sys_stdout = sys.stdout
 
61
 
62
  from inference.flovd_demo import generate_video
63
 
64
+ @spaces.GPU
65
  def run_inference(prompt, image, pose_type, speed, use_flow_integration, cam_pose_name):
66
  log_buffer = io.StringIO()
67
  sys_stdout = sys.stdout