anonymousatom commited on
Commit
850c776
·
verified ·
1 Parent(s): f8169c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -2,9 +2,9 @@ import streamlit as st
2
  from PIL import Image
3
  from ultralytics import YOLO
4
 
5
- st.title("Object Detection with YOLO")
6
 
7
- model = YOLO('best.pt')
8
 
9
  def detect_objects(image):
10
  result = model(image)
 
2
  from PIL import Image
3
  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)