API Bot commited on
Commit
da14c1e
·
1 Parent(s): 6b167ed

Add @spaces.GPU decorator to process_video_api to prevent timeout

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -631,6 +631,7 @@ def create_annotation_preview(video_file, annotations):
631
  return combined
632
 
633
 
 
634
  def process_video_api(
635
  video_file,
636
  annotations_json_str: str,
 
631
  return combined
632
 
633
 
634
+ @spaces.GPU(duration=120) # Allocate GPU for up to 2 minutes
635
  def process_video_api(
636
  video_file,
637
  annotations_json_str: str,