klein-4b correct

app.py

@@ -1,15 +1,6 @@
 import os
 import subprocess
 import sys
-
-subprocess.run([sys.executable, "-m", "pip", "uninstall", "-y", "diffusers"], check=False)
-subprocess.check_call(["unzip", "-o", "diffusers.zip", "-d", "diffusers_local"])
-subprocess.check_call([sys.executable, "-m", "pip", "install", "--no-cache-dir", "./diffusers_local"])
-
-subprocess.check_call(["pwd"])
-subprocess.check_call(["ls", "-la"])
-
-
 import io
 import gradio as gr
 import numpy as np
@@ -72,19 +63,19 @@ pipe_distilled.to("cuda")
 # Dictionary for easy access
 pipes = {
     "Distilled (4 steps)": pipe_distilled,
-    "Regular (30 steps)": pipe_regular,
+    "Base (50 steps)": pipe_regular,
 }
 
 # Default steps for each mode
 DEFAULT_STEPS = {
     "Distilled (4 steps)": 4,
-    "Regular (30 steps)": 30,
+    "Base (50 steps)": 50,
 }
 
 # Default CFG for each mode
 DEFAULT_CFG = {
     "Distilled (4 steps)": 1.0,
-    "Regular (30 steps)": 4.0,
+    "Base (50 steps)": 4.0,
 }
 
 def image_to_data_uri(img):
@@ -267,7 +258,7 @@ FLUX.2 [Klein] is a... [[model](https://huggingface.co/black-forest-labs/FLUX.2-
 
                 mode_choice = gr.Radio(
                     label="Mode",
-                    choices=["Distilled (4 steps)", "Regular (30 steps)"],
+                    choices=["Distilled (4 steps)", "Base (50 steps)"],
                     value="Distilled (4 steps)",
                 )