Update app.py

app.py

@@ -18,6 +18,8 @@ import numpy as np
       
 #       return im_rgb
 
+label = {0:'grenade', 1: 'knife', 2: 'pistol', 3: 'rifle'}
+
 def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
              
     model = YOLO('best (1).pt')
@@ -34,7 +36,7 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
         conf = r.boxes.conf
         cls = r.boxes.cls
         
-        text += (f"Detected {cls} with confidence {conf}\n")
+        text += (f"Detected {label[int(cls)]]} with confidence {double(conf)}\n")
             
     return render,text