Dricz commited on
Commit
599b5b5
1 Parent(s): 381f849

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,6 +10,7 @@ from io import BytesIO
10
  import os
11
 
12
  model = YOLO('best (1).pt')
 
13
  name = ['grenade','knife','pistol','rifle']
14
  image_directory = "/home/user/app/image"
15
  video_directory = "/home/user/app/video"
@@ -87,8 +88,7 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
87
  im_array = r.plot()
88
  im = Image.fromarray(im_array[..., ::-1])
89
 
90
-
91
- model2 = pipeline('image-classification','Kaludi/csgo-weapon-classification')
92
  weapon_name, text_detection = response(image)
93
 
94
 
 
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"
16
  video_directory = "/home/user/app/video"
 
88
  im_array = r.plot()
89
  im = Image.fromarray(im_array[..., ::-1])
90
 
91
+
 
92
  weapon_name, text_detection = response(image)
93
 
94