akhaliq HF Staff commited on
Commit
e8dc09b
·
1 Parent(s): 2b2aefb

Switch to size='xlarge' GPU pool to route past faulty hardware ECC nodes

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -362,7 +362,7 @@ SAFETY_MARGIN = 25
362
  EST_SECONDS_PER_FRAME = 20
363
 
364
 
365
- @spaces.GPU(duration=120)
366
  def run_image_gpu_api(
367
  image_path: str, category: str, model_mode: str, temp: float, top_p: float, top_k: int,
368
  short_size: int | None, question_override: str | None
@@ -410,7 +410,7 @@ def run_image_gpu_api(
410
  return out_img_path, stats, output_text, detections_summary
411
 
412
 
413
- @spaces.GPU(duration=240)
414
  def run_video_gpu_api(
415
  video_path: str, category: str, model_mode: str, temp: float, top_p: float, top_k: int,
416
  short_size: int | None, question_override: str | None, max_video_frames: int
 
362
  EST_SECONDS_PER_FRAME = 20
363
 
364
 
365
+ @spaces.GPU(duration=120, size="xlarge")
366
  def run_image_gpu_api(
367
  image_path: str, category: str, model_mode: str, temp: float, top_p: float, top_k: int,
368
  short_size: int | None, question_override: str | None
 
410
  return out_img_path, stats, output_text, detections_summary
411
 
412
 
413
+ @spaces.GPU(duration=240, size="xlarge")
414
  def run_video_gpu_api(
415
  video_path: str, category: str, model_mode: str, temp: float, top_p: float, top_k: int,
416
  short_size: int | None, question_override: str | None, max_video_frames: int