Update app.py

app.py

@@ -273,7 +273,8 @@ def generate_30(
     num_inference_steps: int = 125,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True, 
-    num_images: int = 1,  
+    num_images: int = 1,
+    strength: float = 0.3,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     torch.backends.cudnn.benchmark = False
@@ -288,6 +289,7 @@ def generate_30(
         "prompt": [prompt] * num_images,
         "negative_prompt": [negative_prompt],
         "negative_prompt_2": [neg_prompt_2],
+        "strength": denoise,
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -339,6 +341,7 @@ def generate_60(
     randomize_seed: bool = False,
     use_resolution_binning: bool = True, 
     num_images: int = 1,
+    strength: float = 0.3,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     torch.backends.cudnn.benchmark = True
@@ -353,6 +356,7 @@ def generate_60(
         "prompt": [prompt] * num_images,
         "negative_prompt": [negative_prompt],
         "negative_prompt_2": [neg_prompt_2],
+        "strength": denoise,
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -403,7 +407,8 @@ def generate_90(
     num_inference_steps: int = 250,
     randomize_seed: bool = False,
     use_resolution_binning: bool = True, 
-    num_images: int = 1,  
+    num_images: int = 1,
+    strength: float = 0.3,
     progress=gr.Progress(track_tqdm=True)  # Add progress as a keyword argument
 ):
     torch.backends.cudnn.benchmark = True
@@ -418,6 +423,7 @@ def generate_90(
         "prompt": [prompt] * num_images,
         "negative_prompt": [negative_prompt],
         "negative_prompt_2": [neg_prompt_2],
+        "strength": denoise,
         "width": width,
         "height": height,
         "guidance_scale": guidance_scale,
@@ -561,6 +567,13 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
                 step=0.1,
                 value=3.8,
             )
+            denoise = gr.Slider(
+                label="Denoise Strength",
+                minimum=0.0,
+                maximum=1.0,
+                step=0.01,
+                value=0.3,
+            )
             num_inference_steps = gr.Slider(
                 label="Number of inference steps",
                 minimum=10,
@@ -601,6 +614,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             randomize_seed, 
             num_images, 
+            denoise
         ],
         outputs=[result, seed],
     )
@@ -624,6 +638,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             randomize_seed, 
             num_images, 
+            denoise
         ],
         outputs=[result, seed],
     )
@@ -647,6 +662,7 @@ with gr.Blocks(theme=gr.themes.Origin(),css=css) as demo:
             num_inference_steps,
             randomize_seed, 
             num_images, 
+            denoise
         ],
         outputs=[result, seed],
     )