Update app.py

app.py

@@ -119,15 +119,28 @@ def app():
         with gr.Row():
             with gr.Column():
                 gr.HTML("<h2>Input Parameters</h2>")
-                img_path = gr.Image(label="Image", height = 370, width = 600)
-                vid_path = gr.Video(label="Video", height = 370, width = 600)
+                img_path = gr.Image(label="Image", height = 260, width = 410)
+                vid_path = gr.Video(label="Video", height = 260, width = 410)
+                #gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
+                #gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
+            with gr.Column(min_width = 270):
+                gr.HTML("<h2>Output</h2>")
+                output_image = gr.Image(type="numpy",label="Output", height = 260, width = 410)
+                #df = gr.BarPlot(show_label=False, x="label", y="counts", x_title="Labels", y_title="Counts", vertical=False)
+                output_video = gr.Video(label="Output", height = 260, width = 410) 
+                #frame_counts_df = gr.LinePlot(show_label=False, x="frame", y="count", x_title="Frame", y_title="Counts", color="label")
+                fig = gr.Plot(label = "label")
+                #output_path = gr.Textbox(label="Output path")
+            with gr.Column():
+                gr.HTML("<h2>Example, Model and Tracking Algorithm</h2>")
                 model_id = gr.Dropdown(
-                    label="Model",
-                    choices=[
-                        "Our_Model.pt",
-                        "yolov9_e_trained.pt"
-                    ],
-                    value="Our_Model.pt"
+                label="Model",
+                choices=[
+                    "our-converted.pt",
+                    "last_best_model.pt"
+                ],
+                value="our-converted.pt"
+                
                 )
                 tracking_algorithm = gr.Dropdown(
                     label= "Tracking Algorithm",
@@ -139,16 +152,9 @@ def app():
                     value="None"
                 )
                 yolov9_infer = gr.Button(value="Inference")
-                gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False)
-                gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False)
-            with gr.Column():
-                gr.HTML("<h2>Output</h2>")
-                output_image = gr.Image(type="numpy",label="Output")
-                #df = gr.BarPlot(show_label=False, x="label", y="counts", x_title="Labels", y_title="Counts", vertical=False)
-                output_video = gr.Video(label="Output") 
-                #frame_counts_df = gr.LinePlot(show_label=False, x="frame", y="count", x_title="Frame", y_title="Counts", color="label")
-                fig = gr.Plot(label = "Plot")
-                #output_path = gr.Textbox(label="Output path")
+                gr.Examples(['./img_examples/Exam_1.png','./img_examples/Exam_2.png','./img_examples/Exam_3.png','./img_examples/Exam_4.png','./img_examples/Exam_5.png'], inputs=img_path,label = "Image Example", cache_examples = False, examples_per_page = 3)
+                gr.Examples(['./video_examples/video_1.mp4', './video_examples/video_2.mp4','./video_examples/video_3.mp4','./video_examples/video_4.mp4','./video_examples/video_5.mp4'], inputs=vid_path, label = "Video Example", cache_examples = False, examples_per_page = 3)
+
                 
 
         yolov9_infer.click(
@@ -162,7 +168,6 @@ def app():
             outputs=[output_image, output_video, fig],
         )
 
-
 gradio_app = gr.Blocks()
 with gradio_app:
     gr.HTML(