Fifthoply commited on
Commit
020b103
·
verified ·
1 Parent(s): a1b4ce5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -257,7 +257,7 @@ def _estimate_gpu_duration(images: torch.Tensor, num_scale_frames: int, keyframe
257
  return min(180, max(60, 24 + frame_count * 4))
258
 
259
 
260
- @spaces.GPU(duration=_estimate_gpu_duration)
261
  def _run_inference(images: torch.Tensor, num_scale_frames: int, keyframe_interval: int) -> tuple[dict[str, Any], torch.Tensor, dict[str, Any]]:
262
  bundle = _load_model_bundle()
263
  model = bundle["model"]
 
257
  return min(180, max(60, 24 + frame_count * 4))
258
 
259
 
260
+ @spaces.GPU(duration=420)
261
  def _run_inference(images: torch.Tensor, num_scale_frames: int, keyframe_interval: int) -> tuple[dict[str, Any], torch.Tensor, dict[str, Any]]:
262
  bundle = _load_model_bundle()
263
  model = bundle["model"]