anonymousatom commited on
Commit
5a6db50
·
verified ·
1 Parent(s): b5c677d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from ultralytics import YOLO
4
 
5
  st.title("Gun/Arms Detection")
6
 
7
- model = YOLO('yolov8n_best.pt')
8
 
9
  def detect_objects(image):
10
  result = model(image, conf = 0.55)
 
4
 
5
  st.title("Gun/Arms Detection")
6
 
7
+ model = YOLO('yolov8_background1k_best.pt')
8
 
9
  def detect_objects(image):
10
  result = model(image, conf = 0.55)