Update app.py

app.py

@@ -42,6 +42,7 @@ def invert_and_edit(image,
                     start_timestep, 
                     stop_timestep, 
                     num_inversion_steps,
+                    num_inference_steps,
                     width,
                     height,
                     inverted_latents,
@@ -64,7 +65,7 @@ def invert_and_edit(image,
     latent_image_ids=latent_image_ids.to(DEVICE),
     start_timestep=start_timestep, 
     stop_timestep=stop_timestep,
-    num_inference_steps=num_inversion_steps,
+    num_inference_steps=num_inference_steps,
     eta=eta, 
     ).images[0]
         
@@ -148,6 +149,13 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
             )
             
             randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+            num_inference_steps = gr.Slider(
+                        label="num inference steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=28,
+                    )
 
             with gr.Row():
                 gamma = gr.Slider(
@@ -163,7 +171,7 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
                         minimum=1,
                         maximum=50,
                         step=1,
-                        value=8,
+                        value=28,
                     )
             
             with gr.Row():
@@ -195,6 +203,7 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
             start_timestep, 
             stop_timestep, 
             num_inversion_steps,
+            num_inference_steps,
             width,
             height,
             inverted_latents,
@@ -219,5 +228,10 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
         outputs=[do_inversion]
     )
 
+    num_inversion_steps.change(
+        fn=reset_do_inversion,
+        outputs=[do_inversion]
+    )
+
 if __name__ == "__main__":
     demo.launch()