Spaces:
Build error
Build error
fix recursion
Browse files
src/gradio_space_ci/webhook.py
CHANGED
|
@@ -83,9 +83,11 @@ def enable_space_ci() -> None:
|
|
| 83 |
config = card.data.get("space_ci", {})
|
| 84 |
print(f"Enabling Space CI with config from README: {config}")
|
| 85 |
|
|
|
|
|
|
|
| 86 |
@wraps(gr.Blocks.launch)
|
| 87 |
-
def
|
| 88 |
-
configure_space_ci(
|
| 89 |
blocks=self,
|
| 90 |
trusted_authors=config.get("trusted_authors"),
|
| 91 |
private=config.get("private", "auto"),
|
|
@@ -93,10 +95,13 @@ def enable_space_ci() -> None:
|
|
| 93 |
secrets=config.get("secrets"),
|
| 94 |
hardware=config.get("hardware"),
|
| 95 |
storage=config.get("storage"),
|
| 96 |
-
)
|
|
|
|
|
|
|
|
|
|
| 97 |
|
| 98 |
# Monkey patch gradio
|
| 99 |
-
gr.Blocks.launch =
|
| 100 |
|
| 101 |
|
| 102 |
def configure_space_ci(
|
|
|
|
| 83 |
config = card.data.get("space_ci", {})
|
| 84 |
print(f"Enabling Space CI with config from README: {config}")
|
| 85 |
|
| 86 |
+
old_launch = gr.Blocks.launch
|
| 87 |
+
|
| 88 |
@wraps(gr.Blocks.launch)
|
| 89 |
+
def new_launch(self: gr.Blocks, *args, **kwargs) -> None:
|
| 90 |
+
server = configure_space_ci(
|
| 91 |
blocks=self,
|
| 92 |
trusted_authors=config.get("trusted_authors"),
|
| 93 |
private=config.get("private", "auto"),
|
|
|
|
| 95 |
secrets=config.get("secrets"),
|
| 96 |
hardware=config.get("hardware"),
|
| 97 |
storage=config.get("storage"),
|
| 98 |
+
)
|
| 99 |
+
# De-monkey patch gradio (otherwise it will be called recursively)
|
| 100 |
+
gr.Blocks.launch = old_launch
|
| 101 |
+
server.launch(*args, *kwargs)
|
| 102 |
|
| 103 |
# Monkey patch gradio
|
| 104 |
+
gr.Blocks.launch = new_launch
|
| 105 |
|
| 106 |
|
| 107 |
def configure_space_ci(
|