Hugging Face's logo

Spaces:
Dricz
/
Weapon_Detection_YOLOv8-4
Runtime error

App Files Community
1
Dricz commited on
Commit
f896bab
·
verified ·
1 Parent(s): ab81131

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -9,7 +9,7 @@ import requests
9
  from io import BytesIO
10
  import os
11
 
12
- model = YOLO('best (1).pt')
13
  model2 = pipeline('image-classification','Kaludi/csgo-weapon-classification')
14
  name = ['grenade','knife','pistol','rifle']
15
  image_directory = "/home/user/app/image"
 
9
  from io import BytesIO
10
  import os
11
 
12
+ model = YOLO('80epoch.pt')
13
  model2 = pipeline('image-classification','Kaludi/csgo-weapon-classification')
14
  name = ['grenade','knife','pistol','rifle']
15
  image_directory = "/home/user/app/image"