Spaces:

Dricz
/

Weapon_Detection_YOLOv8

Paused

Dricz commited on Mar 19

Commit

7a8c28d

•

1 Parent(s): eeb99c1

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -7,7 +7,7 @@ import numpy as np
 from transformers import pipeline
 model = YOLO('best (1).pt')
-model2 = pipeline('Kaludi/csgo-weapon-classification')
 name = ['grenade','knife','pistol','rifle']
  # for i, r in enumerate(results):

 from transformers import pipeline
 model = YOLO('best (1).pt')
+model2 = pipeline('image-classification','Kaludi/csgo-weapon-classification')
 name = ['grenade','knife','pistol','rifle']
  # for i, r in enumerate(results):