button height

app.py

@@ -13,9 +13,6 @@ from huggingface_hub import list_repo_files, hf_hub_download
 from ultralytics import YOLO
 
 import cv2
-import os
-
-print(os.path.abspath(os.getcwd()))
 
 # ---------------------------------
 # 0. Get dataset file names
@@ -165,7 +162,7 @@ with gr.Blocks() as demo:
     with gr.Row():
         dropdown = gr.Dropdown( ["train", "val", "test"], label="Dataset split to use", value="train")
         dataset_choice = gr.Text(label="Using Dataset")   
-    with gr.Row():
+    with gr.Row(equal_height=True):
         index_input = gr.Number(label="Enter image number to display: ", value=0, precision=0)
         go_btn = gr.Button("Apply")
     with gr.Row():