John2J commited on
Commit
c728507
·
verified ·
1 Parent(s): 6ed52a0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -240,7 +240,7 @@ def preprocess_for_removal(images, masks):
240
  MAX_PROC_SIDE = None # None = native resolution; set e.g. 1280 if you hit OOM
241
 
242
  @spaces.GPU(duration=300) # 1080p is heavy — give it time
243
- def inference_and_return_video(dilation_iterations, num_inference_steps, video_state):
244
  if video_state.get("video_path") is None:
245
  video_state["video_path"] = video_path
246
  if video_state["origin_images"] is None or video_state["masks"] is None:
 
240
  MAX_PROC_SIDE = None # None = native resolution; set e.g. 1280 if you hit OOM
241
 
242
  @spaces.GPU(duration=300) # 1080p is heavy — give it time
243
+ def inference_and_return_video(dilation_iterations, num_inference_steps, video_path, video_state):
244
  if video_state.get("video_path") is None:
245
  video_state["video_path"] = video_path
246
  if video_state["origin_images"] is None or video_state["masks"] is None: