Son Ngo commited on
Commit
925abef
·
1 Parent(s): f059a2e

yolo conflict

Browse files
Files changed (2) hide show
  1. app.py +3 -3
  2. requirements.txt +1 -1
app.py CHANGED
@@ -2,10 +2,10 @@ import streamlit as st
2
  import cv2
3
  from PIL import Image
4
  import numpy as np
5
- from yolov5 import YOLOv5
6
 
7
- # Load YOLOv5 model (you can load YOLOv8 similarly)
8
- model = YOLOv5('yolo/best.pt')
9
 
10
  st.title("YOLO Object Detection Web App")
11
 
 
2
  import cv2
3
  from PIL import Image
4
  import numpy as np
5
+ from ultralytics import YOLO
6
 
7
+ # Load YOLO model (latest version)
8
+ model = YOLO('yolo/best.pt')
9
 
10
  st.title("YOLO Object Detection Web App")
11
 
requirements.txt CHANGED
@@ -1,4 +1,4 @@
1
  cv2
2
  Pillow
3
  numpy
4
- yolov5
 
1
  cv2
2
  Pillow
3
  numpy
4
+ ultraalytics