Update app.py

app.py

@@ -214,16 +214,16 @@ def app():
                 #         <li><span style="color:#9F33FF">■</span> Truck</li>
                 #     </ul>
                 # """)
-                gr.HTML("""
-                    <h2>Classes (Color)</h2>
-                    <ul>
-                        <li style="font-size:17px;"><span style="color:#FF3333">■</span> Bus</li>
-                        <li style="font-size:17px;"><span style="color:#3358FF">■</span> Bike</li>
-                        <li style="font-size:17px;"><span style="color:#33FF33">■</span> Car</li>
-                        <li style="font-size:17px;"><span style="color:#F6FF33">■</span> Pedestrian</li>
-                        <li style="font-size:17px;"><span style="color:#9F33FF">■</span> Truck</li>
-                    </ul>
-                """)
+                # gr.HTML("""
+                #     <h2>Classes (Color)</h2>
+                #     <ul>
+                #         <li style="font-size:17px;"><span style="color:#FF3333">■</span> Bus</li>
+                #         <li style="font-size:17px;"><span style="color:#3358FF">■</span> Bike</li>
+                #         <li style="font-size:17px;"><span style="color:#33FF33">■</span> Car</li>
+                #         <li style="font-size:17px;"><span style="color:#F6FF33">■</span> Pedestrian</li>
+                #         <li style="font-size:17px;"><span style="color:#9F33FF">■</span> Truck</li>
+                #     </ul>
+                # """)
                 #gr.Image(value = img, interactive = False, label = "Classes", 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)
@@ -256,6 +256,15 @@ def app():
                     value="None"
                 )
                 yolov9_infer = gr.Button(value="Inference")
+                gr.HTML("""
+                    <p style="text-align:center; font-family:Arial; font-size:16px;">
+                        <span style="display:inline-block; width:8px; height:8px; background:#FF3333;"></span> Bus &nbsp;
+                        <span style="display:inline-block; width:8px; height:8px; background:#3358FF;"></span> Bike &nbsp;
+                        <span style="display:inline-block; width:8px; height:8px; background:#33FF33;"></span> Car &nbsp;
+                        <span style="display:inline-block; width:8px; height:8px; background:#F6FF33;"></span> Pedestrian &nbsp;
+                        <span style="display:inline-block; width:8px; height:8px; background:#9F33FF;"></span> Truck
+                    </p>
+                """)
                 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 = 4)
                 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 = 4)