AI-RESEARCHER-2024 commited on
Commit
ffbc3b8
·
verified ·
1 Parent(s): 5947f32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import os
5
  from ultralytics import YOLO
6
 
7
  # Load the trained model
8
- model = YOLOv10('best.pt')
9
 
10
  # Define class names and colors
11
  class_names = ['IHC', 'OHC-1', 'OHC-2', 'OHC-3']
 
5
  from ultralytics import YOLO
6
 
7
  # Load the trained model
8
+ model = YOLO('best.pt')
9
 
10
  # Define class names and colors
11
  class_names = ['IHC', 'OHC-1', 'OHC-2', 'OHC-3']