Spaces:
Running on Zero
Running on Zero
Update app.py
Browse files
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=
|
| 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"]
|