Atulit23 commited on
Commit
40e7a36
·
verified ·
1 Parent(s): 6989ecb

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. app.py +2 -2
  2. best (14).pt +3 -0
app.py CHANGED
@@ -6,7 +6,7 @@ import matplotlib.patches as patches
6
  import numpy as np
7
  import gradio as gr
8
 
9
- model = YOLO('best (13).pt')
10
 
11
  def index(img_url):
12
  response = requests.get(img_url, stream=True)
@@ -17,7 +17,7 @@ def index(img_url):
17
 
18
  classes_ = {0: 'noti', 1: 'pop'}
19
 
20
- results = model.predict(source=img, conf = 0.7)
21
 
22
  boxes = results[0].boxes.xyxy.tolist()
23
  classes = results[0].boxes.cls.tolist()
 
6
  import numpy as np
7
  import gradio as gr
8
 
9
+ model = YOLO('best (14).pt')
10
 
11
  def index(img_url):
12
  response = requests.get(img_url, stream=True)
 
17
 
18
  classes_ = {0: 'noti', 1: 'pop'}
19
 
20
+ results = model.predict(source=img, conf = 0.6)
21
 
22
  boxes = results[0].boxes.xyxy.tolist()
23
  classes = results[0].boxes.cls.tolist()
best (14).pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3b63d7b6830933cbc56ee3da06eee672b66d5293bb7941374a87818125f0e414
3
+ size 22519641