Update app.py

app.py

@@ -18,7 +18,7 @@ import numpy as np
       
 #       return im_rgb
 
-label = {0:'grenade', 1: 'knife', 2: 'pistol', 3: 'rifle'}
+label = ['grenade','knife','pistol','rifle']
 
 def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold: gr.Slider = 0.3, iou_threshold: gr.Slider = 0.6):
              
@@ -39,7 +39,7 @@ def response2(image: gr.Image = None,image_size: gr.Slider = 640, conf_threshold
         x = xywh[0]
         y = xywh[1]
         
-        text += (f"Detected {label["cls"]} with confidence {double(conf)} at {x}, {y}\n")
+        text += (f"Detected {label[cls]} with confidence {double(conf)} at {x}, {y}\n")
             
     return render,text