Treat a value of 0 as default

app.py

@@ -34,6 +34,7 @@ def load_pipeline(model):
     )
 
 
+logger.debug(f"Loading pipeline: {dict(model=model)}")
 pipe = load_pipeline(DEFAULT_MODEL).to("cuda")
 
 
@@ -57,12 +58,17 @@ def infer(
     # Downscale the image
     init_image.thumbnail((1024, 1024))
 
-    logger.debug(f"Generating image: {dict(prompt=prompt)}")
-    additional_args = (
-        {}
-        if model == "timbrooks/instruct-pix2pix" or strength == 0
-        else dict(strength=strength)
-    )
+    additional_args = {
+        k: v
+        for k, v in dict(
+            strength=strength,
+            num_inference_steps=num_inference_steps,
+            guidance_scale=guidance_scale,
+        ).items()
+        if v
+    }
+
+    logger.debug(f"Generating image: {dict(prompt=prompt, **additional_args)}")
 
     images = pipe(
         prompt=prompt,
@@ -70,8 +76,6 @@ def infer(
         negative_prompt=negative_prompt,
         width=width,
         height=height,
-        num_inference_steps=num_inference_steps,
-        guidance_scale=guidance_scale,
         **additional_args,
     ).images
     return images[0]
@@ -139,10 +143,10 @@ with gr.Blocks(css=css) as demo:
 
                 num_inference_steps = gr.Slider(
                     label="Number of inference steps",
-                    minimum=1,
+                    minimum=0,
                     maximum=100,
                     step=1,
-                    value=50,
+                    value=0,
                 )
 
                 guidance_scale = gr.Slider(