Update app.py

app.py

@@ -136,7 +136,8 @@ with gr.Blocks(theme=get_theme(), title="DeepFakeAI 1.0.0") as ui:
         image_button = gr.Button("START")
         clear_button = gr.ClearButton(value="CLEAR")
         image_output = gr.Image(label="OUTPUT")
-        clear_button.add(image_output)
+        with gr.box():
+            clear_button.add(image_output)
 
         image_button.click(
             run,