Spaces:
Running on Zero
Running on Zero
Switch to size='xlarge' GPU pool to route past faulty hardware ECC nodes
Browse files
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
|