Lazy-cache examples; default inference params; rename to Lens

README.md

@@ -1,5 +1,5 @@
 ---
-title: Microsoft Lens
+title: Lens
 emoji: 🔍
 colorFrom: yellow
 colorTo: red

app.py

@@ -58,19 +58,22 @@ def model_defaults(model_name: str):
 @spaces.GPU(duration=120)
 def generate(
     prompt: str,
-    model_name: str,
-    base_resolution: int,
-    aspect_ratio: str,
-    steps: int,
-    cfg: float,
-    seed: int,
-    randomize_seed: bool,
+    model_name: str = MODEL_CHOICES[0],
+    base_resolution: int = 1024,
+    aspect_ratio: str = "1:1",
+    steps: int | None = None,
+    cfg: float | None = None,
+    seed: int = 0,
+    randomize_seed: bool = True,
     progress=gr.Progress(track_tqdm=True),
 ):
     if not prompt or not prompt.strip():
         raise gr.Error("Please enter a prompt.")
 
     pipe = PIPES[model_name]
+    default_steps, default_cfg = model_defaults(model_name)
+    steps = default_steps if steps is None else int(steps)
+    cfg = default_cfg if cfg is None else float(cfg)
 
     if randomize_seed:
         seed = random.randint(0, MAX_SEED)
@@ -81,8 +84,8 @@ def generate(
         prompt=prompt.strip(),
         base_resolution=int(base_resolution),
         aspect_ratio=aspect_ratio,
-        num_inference_steps=int(steps),
-        guidance_scale=float(cfg),
+        num_inference_steps=steps,
+        guidance_scale=cfg,
         num_images_per_prompt=1,
         generator=generator,
     )
@@ -152,6 +155,10 @@ with gr.Blocks(theme=gr.themes.Citrus(), css=CSS, title="Lens / Lens-Turbo") as
                 ["A green iguana basking on a moss-covered log in a tropical rainforest, every scale rendered sharply, dewdrops on its skin, National Geographic style", MODEL_CHOICES[1]],
             ],
             inputs=[prompt, model],
+            outputs=[image, used_seed],
+            fn=generate,
+            cache_examples=True,
+            cache_mode="lazy",
         )
 
     def _sync_defaults(model_name):