Update app.py

app.py

@@ -49,6 +49,7 @@ def generate(
     height: int = 1024,
     guidance_scale_base: float = 5.0,
     num_inference_steps_base: int = 25,
+    strength_img2img: float = 0.7
     use_vae: bool = False,
     use_lora: bool = False,
     model = 'stabilityai/stable-diffusion-xl-base-1.0',
@@ -117,7 +118,7 @@ def generate(
         images = pipe(
             prompt=prompt,
             image=init_image,
-            strength=0.7,
+            strength=strength_img2img,
             negative_prompt=negative_prompt,
             prompt_2=prompt_2,
             negative_prompt_2=negative_prompt_2,
@@ -232,6 +233,15 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
                 step=1,
                 value=25,
             )
+        with gr.Row():
+            strength_img2img = gr.Slider(
+                info="Strength for Img2Img",
+                label="Strength",
+                minimum=0,
+                maximum=1,
+                step=0.01,
+                value=0.7,
+            )
 
     gr.Examples(
         examples=examples,
@@ -312,6 +322,7 @@ with gr.Blocks(theme=gr.themes.Soft(), css="style.css") as demo:
             height,
             guidance_scale_base,
             num_inference_steps_base,
+            strength_img2img,
             use_vae,
             use_lora,
             model,