Spaces:
Running
on
Zero
Running
on
Zero
Add type hints to infer for MCP compatibility
Browse filesGradio uses type hints to generate MCP schema (at /gradio_api/mcp/schema) that MCP clients use to get tool desc and arg type and desc
app.py
CHANGED
@@ -25,16 +25,16 @@ MAX_IMAGE_SIZE = 1024
|
|
25 |
|
26 |
@spaces.GPU(duration=65)
|
27 |
def infer(
|
28 |
-
prompt,
|
29 |
-
negative_prompt="",
|
30 |
-
seed=42,
|
31 |
-
randomize_seed=False,
|
32 |
-
width=1024,
|
33 |
-
height=1024,
|
34 |
-
guidance_scale=1.5,
|
35 |
-
num_inference_steps=8,
|
36 |
-
progress=gr.Progress(track_tqdm=True),
|
37 |
-
):
|
38 |
"""Generate an image using Stable Diffusion 3.5 Large TurboX.
|
39 |
|
40 |
Args:
|
|
|
25 |
|
26 |
@spaces.GPU(duration=65)
|
27 |
def infer(
|
28 |
+
prompt: str,
|
29 |
+
negative_prompt: str = "",
|
30 |
+
seed: int = 42,
|
31 |
+
randomize_seed: bool = False,
|
32 |
+
width: int = 1024,
|
33 |
+
height: int = 1024,
|
34 |
+
guidance_scale: float = 1.5,
|
35 |
+
num_inference_steps: int = 8,
|
36 |
+
progress: gr.Progress = gr.Progress(track_tqdm=True),
|
37 |
+
) -> tuple["PIL.Image.Image", int]:
|
38 |
"""Generate an image using Stable Diffusion 3.5 Large TurboX.
|
39 |
|
40 |
Args:
|