expose prompt + LoRA strength + steps/guidance for dial-in

app.py

@@ -7,7 +7,7 @@ from diffusers import Flux2KleinPipeline
 
 BASE = "black-forest-labs/FLUX.2-klein-4B"
 LORA = "noema-art/impresstation-klein"
-# Fixed edit instruction for klein (an edit model): trigger words + style + directive.
+# Default edit instruction for klein (an edit model): trigger words + style + directive.
 PROMPT = (
     "stationthis, stationthis, low poly playstation screenshot style. "
     "Convert this image into a low-poly PlayStation-era 3D game screenshot in the stationthis style."
@@ -22,15 +22,16 @@ pipe.to("cuda")
 
 
 @spaces.GPU(duration=120)
-def station_this(image, steps, guidance, seed, randomize, progress=gr.Progress(track_tqdm=True)):
+def station_this(image, prompt, strength, steps, guidance, seed, randomize, progress=gr.Progress(track_tqdm=True)):
     if image is None:
         raise gr.Error("Drop in an image first.")
     if randomize:
         seed = random.randint(0, MAX_SEED)
+    pipe.set_adapters("stationthis", adapter_weights=float(strength))
     g = torch.Generator("cuda").manual_seed(int(seed))
     out = pipe(
         image=[image.convert("RGB")],
-        prompt=PROMPT,
+        prompt=(prompt.strip() or PROMPT),
         guidance_scale=float(guidance),
         num_inference_steps=int(steps),
         generator=g,
@@ -57,7 +58,9 @@ with gr.Blocks(css=CSS, title="stationthis") as demo:
     with gr.Row():
         with gr.Column():
             inp = gr.Image(label="Your image", type="pil", height=360)
-            with gr.Accordion("Settings", open=False):
+            with gr.Accordion("Dial it in", open=True):
+                prompt = gr.Textbox(label="Prompt", value=PROMPT, lines=3)
+                strength = gr.Slider(0.0, 1.5, value=1.0, step=0.05, label="LoRA strength (PS2 intensity)")
                 steps = gr.Slider(4, 30, value=8, step=1, label="Steps")
                 guidance = gr.Slider(1.0, 6.0, value=2.5, step=0.1, label="Guidance")
                 seed = gr.Slider(0, MAX_SEED, value=42, step=1, label="Seed")
@@ -66,7 +69,7 @@ with gr.Blocks(css=CSS, title="stationthis") as demo:
         with gr.Column():
             out = gr.Image(label="result", height=360)
             used_seed = gr.Number(label="Seed used", interactive=False)
-    go.click(station_this, [inp, steps, guidance, seed, randomize], [out, used_seed])
+    go.click(station_this, [inp, prompt, strength, steps, guidance, seed, randomize], [out, used_seed])
 
 if __name__ == "__main__":
     demo.launch()