Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -77,7 +77,7 @@ wan_pipe.enable_vram_management(num_persistent_param_in_dit=None)
|
|
77 |
|
78 |
gr_info_duration = 2 # gradio popup information duration
|
79 |
|
80 |
-
|
81 |
def rmbg_mask(video_path, mask_path=None, progress=gr.Progress()):
|
82 |
"""Extract foreground from video, return foreground video path"""
|
83 |
if not video_path:
|
@@ -142,7 +142,7 @@ def rmbg_mask(video_path, mask_path=None, progress=gr.Progress()):
|
|
142 |
gr.Error(error_msg)
|
143 |
return None
|
144 |
|
145 |
-
|
146 |
def video_relighting(fg_video_path, prompt, seed=-1, num_inference_steps=50, video_quality=7,
|
147 |
progress=gr.Progress()):
|
148 |
"""Relighting the foreground video base on the text """
|
|
|
77 |
|
78 |
gr_info_duration = 2 # gradio popup information duration
|
79 |
|
80 |
+
@spaces.GPU
|
81 |
def rmbg_mask(video_path, mask_path=None, progress=gr.Progress()):
|
82 |
"""Extract foreground from video, return foreground video path"""
|
83 |
if not video_path:
|
|
|
142 |
gr.Error(error_msg)
|
143 |
return None
|
144 |
|
145 |
+
@spaces.GPU
|
146 |
def video_relighting(fg_video_path, prompt, seed=-1, num_inference_steps=50, video_quality=7,
|
147 |
progress=gr.Progress()):
|
148 |
"""Relighting the foreground video base on the text """
|