Update app.py

app.py

@@ -15,7 +15,7 @@ css = """
 }
 """
 
-def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
+def generate(prompt, samp_steps, seed, strength, progress=gr.Progress(track_tqdm=True)):
     if seed < 0:
         seed = random.randint(1,999999)
     image = txt2img(
@@ -28,9 +28,9 @@ def generate(prompt, samp_steps, seed, progress=gr.Progress(track_tqdm=True)):
     final_image = img2img(
         prompt,
         upscaled_image,
-        num_inference_steps=int(samp_steps*2),
+        num_inference_steps=int(samp_steps/strength),
         guidance_scale=5,
-        strength=0.8,
+        strength=strength,
         generator=torch.manual_seed(seed),
     ).images[0]
     return [final_image], seed
@@ -58,12 +58,13 @@ with gr.Blocks(css=css) as demo:
         
         with gr.Row():
             sampling_steps = gr.Slider(1, 10, value=5, step=1, label="Sampling steps")
+            strength = gr.Slider(0, 1, value=0.75, step=0.05, label="Refiner strength")
             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=650)
         
-    submit_btn.click(generate, [prompt, sampling_steps, seed], [gallery, lastSeed], queue=True)
+    submit_btn.click(generate, [prompt, sampling_steps, seed, strength], [gallery, lastSeed], queue=True)
     
 txt2img, img2img = set_base_models()
 demo.launch(debug=True)