Spaces:
Configuration error
Configuration error
Commit
·
d255d9f
1
Parent(s):
87dee1a
gpu timeout
Browse files
app.py
CHANGED
|
@@ -77,7 +77,7 @@ def extract_images_from_zip(zip_path: str, outdir: str) -> list[str]:
|
|
| 77 |
# -------------------------------------------------------------------------
|
| 78 |
# 1) Core model inference
|
| 79 |
# -------------------------------------------------------------------------
|
| 80 |
-
@spaces.GPU
|
| 81 |
def run_model(target_dir: str, model: STream3R, mode: str="causal", streaming: bool=False) -> dict:
|
| 82 |
"""
|
| 83 |
Run the STream3R model on images in the 'target_dir/images' folder and return predictions.
|
|
|
|
| 77 |
# -------------------------------------------------------------------------
|
| 78 |
# 1) Core model inference
|
| 79 |
# -------------------------------------------------------------------------
|
| 80 |
+
@spaces.GPU(duration=180) # triggers ZeroGPU allocation for this call
|
| 81 |
def run_model(target_dir: str, model: STream3R, mode: str="causal", streaming: bool=False) -> dict:
|
| 82 |
"""
|
| 83 |
Run the STream3R model on images in the 'target_dir/images' folder and return predictions.
|