Parthebhan commited on
Commit
934e326
·
verified ·
1 Parent(s): 758f1bb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -0
app.py CHANGED
@@ -1,8 +1,12 @@
1
  import gradio as gr
2
  from ultralytics import YOLO
 
 
3
 
4
  # Define function to perform prediction with YOLOv9 model
5
  def predict_image(image, model_path, image_size, conf_threshold, iou_threshold):
 
 
6
  # Load YOLO model
7
  model = YOLO(model_path)
8
 
 
1
  import gradio as gr
2
  from ultralytics import YOLO
3
+ # Import YOLOv9
4
+ import yolov9
5
 
6
  # Define function to perform prediction with YOLOv9 model
7
  def predict_image(image, model_path, image_size, conf_threshold, iou_threshold):
8
+
9
+
10
  # Load YOLO model
11
  model = YOLO(model_path)
12