Update app.py

app.py

@@ -135,6 +135,9 @@ def yolov9_inference(model_id, img_path=None, vid_path=None, tracking_algorithm
     return output_image, output_video, fig
 
 def app():
+    img = Image.open('./img_examples/classes.png')
+    img = img.resize((410, 260), Image.Resampling.LANCZOS)
+    img = np.array(img)
     with gr.Blocks(title="YOLOv9: Real-time Object Detection", css=".gradio-container {background:lightyellow;}"):
         with gr.Row():
             with gr.Column():
@@ -142,7 +145,7 @@ def app():
                 img_path = gr.Image(label="Image", height = 260, width = 410)
                 vid_path = gr.Video(label="Video", height = 260, width = 410)
                 #gr.HTML("<img src='flie/img_examples/clasess.png'>")
-                gr.Image(value = "./img_examples/clasess.png", interactive = False)
+                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)
             with gr.Column(min_width = 270):