Fixed predict args, minor UI change

app.py

@@ -167,6 +167,7 @@ class ImageEditor(object):
                             "tar_text": tar_text_styleclip,
                             "alpha": alpha_styleclip,
                             "beta": beta_styleclip}
+                            
 
             return func(self, *args, edit_choices)
 
@@ -195,18 +196,17 @@ class ImageEditor(object):
 
     @_pack_edits
     def edit_image(self, input, output_styles, edit_choices):
-        return self.predict(input, output_styles, edit_choices)
+        return self.predict(input, output_styles, edit_choices=edit_choices)
 
     @_pack_edits
     def edit_video(self, input, output_styles, loop_styles, edit_choices):
-        return self.predict(input, output_styles, True, loop_styles, edit_choices)
+        return self.predict(input, output_styles, generate_video=True, loop_styles=loop_styles, edit_choices=edit_choices)
 
     def predict(
         self,
         input,                  # Input image path
         output_styles,          # Style checkbox options.
         generate_video = False, # Generate a video instead of an output image
-        with_editing   = False, # Apply latent space editing to the generated video
         loop_styles    = False, # Loop back to the initial style
         edit_choices   = None,  # Optional dictionary with edit choice arguments
     ):  
@@ -330,16 +330,18 @@ with blocks:
 
     with gr.Tabs():
         with gr.TabItem("Edit Images"):
-            with gr.Column():
-                img_button = gr.Button("Edit Image")
-            with gr.Column():
-                img_output = gr.Gallery(label="Output Images")
+            with gr.Row():
+                with gr.Column():
+                    img_button = gr.Button("Edit Image")
+                with gr.Column():
+                    img_output = gr.Gallery(label="Output Images")
             
         with gr.TabItem("Create Video"):
             with gr.Row():
                 with gr.Column():
-                    vid_button  = gr.Button("Generate Video")
-                    loop_styles = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
+                    with gr.Row():
+                        vid_button  = gr.Button("Generate Video")
+                        loop_styles = gr.inputs.Checkbox(default=True, label="Loop video back to the initial style?")
                     
                 with gr.Column():
                     vid_output = gr.outputs.Video(label="Output Video")