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 """
|