Update app.py
Browse files
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
|