Ryan-PR commited on
Commit
1b2b38c
·
verified ·
1 Parent(s): ef7e314

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -319,7 +319,7 @@ def clear_ref_clicks(ref_state):
319
  return Image.fromarray(ref_state["origin_image"])
320
 
321
 
322
- @spaces.GPU(duration=40)
323
  def track_video(n_frames, video_state):
324
  input_points = video_state["input_points"]
325
  input_labels = video_state["input_labels"]
@@ -398,7 +398,7 @@ def track_video(n_frames, video_state):
398
  return video_file, video_state
399
 
400
 
401
- @spaces.GPU(duration=250)
402
  def inference_and_return_video(
403
  dilate_radius,
404
  num_inference_steps,
 
319
  return Image.fromarray(ref_state["origin_image"])
320
 
321
 
322
+ @spaces.GPU()
323
  def track_video(n_frames, video_state):
324
  input_points = video_state["input_points"]
325
  input_labels = video_state["input_labels"]
 
398
  return video_file, video_state
399
 
400
 
401
+ @spaces.GPU()
402
  def inference_and_return_video(
403
  dilate_radius,
404
  num_inference_steps,