TheKnight115 commited on
Commit
c578959
·
verified ·
1 Parent(s): d641d8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import numpy as np
4
  from ultralytics import YOLO
5
 
6
  # Load the YOLO model
7
- model = YOLO('yolov8_Medium.pt') # Ensure the model file is in the root directory of your Space
8
 
9
  def run_yolo(image):
10
  # Run the model on the image and get results
 
4
  from ultralytics import YOLO
5
 
6
  # Load the YOLO model
7
+ model = YOLO('yolov8_motorbikes_detection_s.pt') # Ensure the model file is in the root directory of your Space
8
 
9
  def run_yolo(image):
10
  # Run the model on the image and get results