Extend ZeroGPU generation lease
Browse files
app.py
CHANGED
|
@@ -59,7 +59,7 @@ def _zero_gpu_entrypoint(fn: Any) -> Any:
|
|
| 59 |
gpu_decorator = getattr(spaces, "GPU", None) if spaces is not None else None
|
| 60 |
if gpu_decorator is None:
|
| 61 |
return fn
|
| 62 |
-
return gpu_decorator(duration=
|
| 63 |
|
| 64 |
|
| 65 |
def _relay_snapshot(adult_mode: bool = False) -> dict[str, Any]:
|
|
|
|
| 59 |
gpu_decorator = getattr(spaces, "GPU", None) if spaces is not None else None
|
| 60 |
if gpu_decorator is None:
|
| 61 |
return fn
|
| 62 |
+
return gpu_decorator(duration=900)(fn)
|
| 63 |
|
| 64 |
|
| 65 |
def _relay_snapshot(adult_mode: bool = False) -> dict[str, Any]:
|