Fly-ShuAI commited on
Commit
feaa9e3
·
verified ·
1 Parent(s): ee1ee3b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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
- # @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,7 +142,7 @@ def rmbg_mask(video_path, mask_path=None, progress=gr.Progress()):
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 """
 
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 """