reduced gpu time

app.py

@@ -2,6 +2,7 @@ from __future__ import annotations
 
 import importlib.util
 import os
+import re
 import runpy
 import shlex
 import shutil
@@ -428,20 +429,24 @@ def patch_upstream_gradio_for_zerogpu(demo_entrypoint: Path) -> None:
             raise RuntimeError(f"Could not insert ZeroGPU import in {demo_entrypoint}")
         source = source.replace("import torch\n", "import spaces\nimport torch\n", 1)
 
-    if "@spaces.GPU(" not in source:
-        if "def process_video(\n" not in source:
-            raise RuntimeError(f"Could not locate process_video in {demo_entrypoint}")
-        source = source.replace(
-            "def process_video(\n",
-            (
-                "@spaces.GPU("
-                f"duration={int(ZEROGPU_DURATION_SECONDS)}, "
-                f"size={ZEROGPU_SIZE!r}"
-                ")\n"
-                "def process_video(\n"
-            ),
-            1,
-        )
+    if "def process_video(\n" not in source:
+        raise RuntimeError(f"Could not locate process_video in {demo_entrypoint}")
+
+    source = re.sub(
+        r"@spaces\.GPU\([^\n]*\)\n(?=def process_video\()",
+        "",
+        source,
+        count=1,
+    )
+
+    desired_decorator = (
+        "@spaces.GPU("
+        f"duration={int(ZEROGPU_DURATION_SECONDS)}, "
+        f"size={ZEROGPU_SIZE!r}"
+        ")\n"
+    )
+    if desired_decorator not in source:
+        source = source.replace("def process_video(\n", f"{desired_decorator}def process_video(\n", 1)
 
     injected_css_loader = (
         "\n"