Update app.py

app.py

@@ -135,7 +135,6 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
         target_image = gr.Image(type="filepath", label="TARGET IMAGE")
         image_button = gr.Button("START")
         clear_button = gr.ClearButton(value="CLEAR")
-        image_output = gr.Image(label="OUTPUT")
         clear_button.add(image_output)
 
         image_button.click(
@@ -143,6 +142,9 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
             inputs=[source_image, target_image, unique_id, frame_processor_checkbox, face_analyser_direction_dropdown, face_analyser_age_dropdown, face_analyser_gender_dropdown],
             outputs=image_output
         )
+        with gr.Box():
+            image_output = gr.Image(label="OUTPUT")
+            
         clear_button.click(fn=clear_output, inputs=unique_id)
 
     with gr.Tab("Video:               "):