Update app.py

app.py

@@ -98,10 +98,6 @@ def get_theme() -> gr.Theme:
         block_label_text_size = '*text_sm',
         block_title_text_size = '*text_sm'
     )
-    
-css='''
-#image_output{max-height:500px}
-'''
 
 with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
     with gr.Box():
@@ -142,13 +138,12 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             with gr.Column():
                 target_image = gr.Image(type="filepath", label="TARGET IMAGE")
         with gr.Row():
-            with gr.Column():
-                image_button = gr.Button("START")
             with gr.Column():
                 clear_button = gr.ClearButton(value="CLEAR")
-        with gr.Row():
             with gr.Column():
-                image_output = gr.Image(elem_id="image_output", label="OUTPUT", height=500)
+                image_button = gr.Button("START")
+        with gr.Row():
+           image_output = gr.Image(label="OUTPUT", height=500)
         clear_button.add(image_output)
 
         image_button.click(
@@ -169,9 +164,9 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             skip_audio = gr.Checkbox(label="SKIP AUDIO")
             keep_fps = gr.Checkbox(label="KEEP FPS")
             keep_temp = gr.Checkbox(label="KEEP TEMP")
-        video_button = gr.Button("START")
         clear_video_button = gr.ClearButton(value="CLEAR")
-        video_output = gr.Video(label="OUTPUT")
+        video_button = gr.Button("START")
+        video_output = gr.Video(label="OUTPUT", height=500)
         clear_video_button.add(video_output)
         video_button.click(
             run,