brian4dwell commited on
Commit
d255d9f
·
1 Parent(s): 87dee1a

gpu timeout

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 # 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.
 
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.