Update app.py
Browse files
app.py
CHANGED
@@ -18,7 +18,7 @@ import numpy as np
|
|
18 |
|
19 |
# return im_rgb
|
20 |
|
21 |
-
|
22 |
|
23 |
def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
|
24 |
|
|
|
18 |
|
19 |
# return im_rgb
|
20 |
|
21 |
+
name = ['grenade','knife','pistol','rifle']
|
22 |
|
23 |
def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
|
24 |
|