Update app.py

app.py

@@ -31,8 +31,18 @@ examples = [
 ]
 
 
-# Build Gradio UI
-with gr.Blocks() as demo:
+# Building Gradio UI
+
+
+custom_theme = gr.themes.Base().set(
+    body_background_fill="#0f0f11",   # background color
+    block_background_fill="#0f0f11",  # blocks background
+    block_border_color="#0f0f11",     # remove border feel
+    background_fill_primary="#0f0f11" # for other sections
+)
+
+
+with gr.Blocks(theme=custom_theme) as demo:
 
     gr.HTML(
     """
@@ -43,6 +53,7 @@ with gr.Blocks() as demo:
         style='height: 100px; width: auto; display: block; margin: 0 auto;'
     >
     <h2>V-Safe: Construction Site Safety Detection Demo</h2>
+    <br>
 </div>
     """
     )
@@ -51,13 +62,13 @@ with gr.Blocks() as demo:
 
     with gr.Row():
         with gr.Column():
-            input_img = gr.Image(type="pil", label="Upload an Image")
+            input_img = gr.Image(type="pil", label="Upload an Image", height=640)
             confidence = gr.Slider(0.0, 1.0, value=0.25, step=0.05, label="Confidence Threshold")
             
             run_btn = gr.Button("Run Inference")
 
         with gr.Column():
-            output_img = gr.Image(type="pil", label="Annotated Result")
+            output_img = gr.Image(type="pil", label="Annotated Result", height=640)
 
     run_btn.click(
         fn=predict,