Update app.py

app.py

@@ -70,7 +70,7 @@ def invert_and_edit(image,
     eta=eta, 
     ).images[0]
         
-    return output, inverted_latents, image_latents, latent_image_ids, do_inversion, seed
+    return output, inverted_latents.value, image_latents.value, latent_image_ids.value, do_inversion, seed
 
 # UI CSS
 css = """
@@ -103,43 +103,70 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
                     label="Input Image",
                     type="pil"
                 )
-                eta = gr.Slider(
-                    label="eta",
-                    info = "lower eta to ehnace the edits",
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.1,
-                    value=0.9,
-                )
                 prompt = gr.Text(
                     label="Edit Prompt",
                     max_lines=1,
                     placeholder="describe the edited output",
                 )
+                with gr.Row():          
+                    eta = gr.Slider(
+                        label="eta",
+                        info = "lower eta to ehnace the edits",
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.1,
+                        value=0.9,
+                    )
+                    start_timestep = gr.Slider(
+                    label="start timestep",
+                    info = "lower gamma to ehnace the edits",
+                    minimum=0.0,
+                    maximum=1.0,
+                    step=0.1,
+                    value=0.9,
+                )
+                    stop_timestep = gr.Slider(
+                        label="stop timestep",
+                        info = "lower gamma to ehnace the edits",
+                        minimum=0.0,
+                        maximum=1.0,
+                        step=0.1,
+                        value=0.9,
+                    )
+                
                 run_button = gr.Button("Edit", variant="primary")
             
             with gr.Column():
                 result = gr.Image(label="Result")
         
         with gr.Accordion("Advanced Settings", open=False):
-            
-            seed = gr.Slider(
-                label="Seed",
-                minimum=0,
-                maximum=MAX_SEED,
-                step=1,
-                value=42,
-            )
-            
-            randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
-
-            num_inversion_steps = gr.Slider(
-                    label="num inversion steps",
-                    minimum=1,
-                    maximum=50,
+            with gr.Row():
+                seed = gr.Slider(
+                    label="Seed",
+                    minimum=0,
+                    maximum=MAX_SEED,
                     step=1,
-                    value=8,
+                    value=42,
+                )
+                
+                randomize_seed = gr.Checkbox(label="Randomize seed", value=True)
+
+            with gr.Row():
+                gamma = gr.Slider(
+                    label="gamma",
+                    info = "lower gamma to ehnace the edits",
+                    minimum=0.0,
+                    maximum=1.0,
+                    step=0.1,
+                    value=0.9,
                 )
+                num_inversion_steps = gr.Slider(
+                        label="num inversion steps",
+                        minimum=1,
+                        maximum=50,
+                        step=1,
+                        value=8,
+                    )
             
             with gr.Row():
                 width = gr.Slider(
@@ -157,31 +184,8 @@ Stochastic Rectified Differential Equations*](https://rf-inversion.github.io/dat
                     step=32,
                     value=1024,
                 )
-            with gr.Row():
-                gamma = gr.Slider(
-                    label="gamma",
-                    info = "lower gamma to ehnace the edits",
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.1,
-                    value=0.9,
-                )
-                start_timestep = gr.Slider(
-                    label="start timestep",
-                    info = "lower gamma to ehnace the edits",
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.1,
-                    value=0.9,
-                )
-                stop_timestep = gr.Slider(
-                    label="stop timestep",
-                    info = "lower gamma to ehnace the edits",
-                    minimum=0.0,
-                    maximum=1.0,
-                    step=0.1,
-                    value=0.9,
-                )
+            
+                
     
     run_button.click(
         fn=invert_and_edit,