Update app.py

app.py

@@ -3,27 +3,26 @@ from ultralytics import YOLO
 # Import YOLOv9
 import yolov9
 
-# Define function to perform prediction with YOLOv9 model
+# Define function to perform prediction with YOLO model
 def predict_image(image, model_path, image_size, conf_threshold, iou_threshold):
 
-    
     # Load YOLO model
     model = YOLO(model_path)
     
     # Perform inference with YOLO model
-    results = model(image, size=image_size, conf=conf_threshold, iou=iou_threshold)
+    results = model.predict(image, size=image_size, conf=conf_threshold, iou=iou_threshold)
     
     # Render the output
     output_image = results.render()
     
-    return output_image[0]
+    return output_image
 
 # Define Gradio interface
 def app():
     with gr.Blocks():
         with gr.Row():
             with gr.Column():
-                img_path = gr.Image(type="filepath", label="Image")
+                img_path = gr.Image(type="file", label="Image")
                 model_path = gr.Dropdown(
                     label="Model",
                     choices=[
@@ -52,10 +51,10 @@ def app():
                     step=0.1,
                     value=0.5,
                 )
-                yolov9_infer = gr.Button(value="Submit")
+                yolov9_infer = gr.Button(label="Submit")
 
             with gr.Column():
-                output_numpy = gr.Image(type="numpy", label="Output")
+                output_image = gr.Image(type="numpy", label="Output")
 
         yolov9_infer.click(
             fn=predict_image,
@@ -66,7 +65,7 @@ def app():
                 conf_threshold,
                 iou_threshold,
             ],
-            outputs=[output_numpy],
+            outputs=[output_image],
         )   
 
 gradio_app = gr.Blocks()