bug fix

app.py

@@ -152,32 +152,18 @@ def patch_upstream_gradio_for_zerogpu(demo_entrypoint: Path) -> None:
             1,
         )
 
-    if "def load_gradio_hero_css():\n" not in source:
-        marker = "GRADIO_HERO_CSS_PATH = ASSETS_CSS_DIR / \"gradio_hero.css\"\n"
-        if marker not in source:
-            raise RuntimeError(f"Could not locate CSS path constant in {demo_entrypoint}")
-        source = source.replace(
-            marker,
-            (
-                marker +
-                "\n"
-                "@lru_cache(maxsize=1)\n"
-                "def load_gradio_hero_css():\n"
-                "    if not GRADIO_HERO_CSS_PATH.exists():\n"
-                "        return None\n"
-                "    return GRADIO_HERO_CSS_PATH.read_text(encoding=\"utf-8\")\n"
-            ),
-            1,
-        )
-
-    source = source.replace("        css=load_gradio_hero_css(),\n    ) as app:\n", "    ) as app:\n")
+    injected_css_loader = (
+        "\n"
+        "@lru_cache(maxsize=1)\n"
+        "def load_gradio_hero_css():\n"
+        "    if not GRADIO_HERO_CSS_PATH.exists():\n"
+        "        return None\n"
+        "    return GRADIO_HERO_CSS_PATH.read_text(encoding=\"utf-8\")\n"
+    )
+    if injected_css_loader in source:
+        source = source.replace(injected_css_loader, "", 1)
 
-    launch_css_marker = "        server_port=args.server_port,\n"
-    launch_css_line = "        css=load_gradio_hero_css(),\n"
-    if launch_css_line not in source:
-        if launch_css_marker not in source:
-            raise RuntimeError(f"Could not locate Gradio launch arguments in {demo_entrypoint}")
-        source = source.replace(launch_css_marker, launch_css_marker + launch_css_line, 1)
+    source = source.replace("        css=load_gradio_hero_css(),\n", "")
 
     demo_entrypoint.write_text(source, encoding="utf-8")