mcp-docstrings-and-type-hints

README.md

@@ -4,7 +4,7 @@ emoji: 🏃
 colorFrom: purple
 colorTo: blue
 sdk: gradio
-sdk_version: 5.30.0
+sdk_version:  5.32.1
 app_file: app.py
 pinned: true
 license: other

app.py

@@ -25,16 +25,31 @@ MAX_IMAGE_SIZE = 1024
 
 @spaces.GPU(duration=65)
 def infer(
-    prompt,
-    negative_prompt="",
-    seed=42,
-    randomize_seed=False,
-    width=1024,
-    height=1024,
-    guidance_scale=1.5,
-    num_inference_steps=8,
-    progress=gr.Progress(track_tqdm=True),
-):
+    prompt: str,
+    negative_prompt: str = "",
+    seed: int = 42,
+    randomize_seed: bool = False,
+    width: int = 1024,
+    height: int = 1024,
+    guidance_scale: float = 1.5,
+    num_inference_steps: int = 8,
+    progress: gr.Progress = gr.Progress(track_tqdm=True),
+) -> tuple["PIL.Image.Image", int]:
+    """Generate an image using Stable Diffusion 3.5 Large TurboX.
+    
+    Args:
+        prompt: The prompt to generate an image from
+        negative_prompt: The negative prompt
+        seed: The generation seed
+        randomize_seed: Whether to randomize the seed
+        width: The width of the generated image in pixels
+        height: The height of the generated image in pixels
+        guidance_scale: How closely the image should align with the prompt
+        num_inference_steps: Number of generation steps
+
+    Returns:
+        The generated image and the seed used for generation as a tuple of (PIL.Image.Image, seed)
+    """
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)