Update app.py

app.py

@@ -16,12 +16,13 @@ css = """
 }
 """
 
-def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, turbo_steps, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
+    print("prompt = ", prompt)
     if seed < 0:
         seed = random.randint(1,999999)
     image = txt2img(
         prompt,
-        num_inference_steps=1,
+        num_inference_steps=turbo_steps,
         guidance_scale=0.0,
         generator=torch.manual_seed(seed),
     ).images[0]
@@ -58,13 +59,14 @@ with gr.Blocks(css=css) as demo:
         submit_btn = gr.Button("Generate", elem_classes="btn-green")
         
         with gr.Row():
+            turbo_steps = gr.Slider(1, 4, value=1, step=1, label="Turbo steps")
             sampling_steps = gr.Slider(1, 6, value=3, step=1, label="Refiner steps")
             seed = gr.Number(label="Seed", value=-1, minimum=-1, precision=0)
             lastSeed = gr.Number(label="Last Seed", value=-1, interactive=False)
             
         gallery = gr.Gallery(show_label=False, preview=True, container=False, height=1100)
         
-    submit_btn.click(generate, [prompt, sampling_steps, seed], [gallery, lastSeed], queue=True)
+    submit_btn.click(generate, [prompt, turbo_steps, sampling_steps, seed], [gallery, lastSeed], queue=True)
     
 txt2img, img2img = set_base_models()
 demo.launch(debug=True)