Update app.py

app.py

@@ -97,6 +97,8 @@ css = """
 .gradio-container {
     max-width: 1400px !important;
     margin: auto;
+    padding-left: 10px !important;
+    padding-right: 10px !important;
 }
 
 footer {
@@ -110,11 +112,17 @@ footer {
     margin-bottom:10px;
 }
 
-.subtitle {
-    text-align:center;
-    font-size:18px;
-    color:#555;
-    margin-bottom:25px;
+@media (max-width: 768px) {
+
+    .main-title {
+        font-size:28px !important;
+        line-height:1.2 !important;
+    }
+
+    .gradio-container {
+        padding-left:5px !important;
+        padding-right:5px !important;
+    }
 }
 """
 
@@ -289,19 +297,18 @@ with gr.Blocks(
 
         with gr.Tab("🖼️ Generate"):
 
-            with gr.Row():
-
+            with gr.Row(equal_height=True):
+            
                 image_input = gr.Image(
-                    sources=[
-                        "upload",
-                        "webcam"
-                    ],
+                    sources=["upload", "webcam"],
                     type="pil",
-                    label="Input Image"
+                    label="Input Image",
+                    height=350
                 )
-
+            
                 image_output = gr.Image(
-                    label="Original Image"
+                    label="Original Image",
+                    height=350
                 )
 
             task_dropdown = gr.Dropdown(
@@ -344,14 +351,16 @@ with gr.Blocks(
                 lines=5
             )
 
-            with gr.Row():
-
+            with gr.Row(equal_height=True):
+            
                 confidence_output = gr.Textbox(
-                    label="Confidence Score"
+                    label="Confidence Score",
+                    scale=1
                 )
-
+            
                 time_output = gr.Textbox(
-                    label="Inference Time"
+                    label="Inference Time",
+                    scale=1
                 )
 
             generate_btn.click(
@@ -429,7 +438,8 @@ with gr.Blocks(
 
             gr.Dataframe(
                 value=metrics_df,
-                interactive=False
+                interactive=False,
+                wrap=True
             )
 
 demo.launch()