Update app.py

app.py

@@ -32,20 +32,20 @@ def process_img(image):
             for element in boxes:
                 box=np.array(element.cpu()).flatten()
                 if result[0].boxes.cls[0].cpu().numpy()==2:
-                        label='car'
+                        lbel='car'
                 if result[0].boxes.cls[0].cpu().numpy()==0:
-                        label='bicycle'
+                        lbel='bicycle'
                 if result[0].boxes.cls[0].cpu().numpy()==1:
-                        label='bus'
+                        lbel='bus'
                 if result[0].boxes.cls[0].cpu().numpy()==3:
-                        label='motorcycle'
+                        lbel='motorcycle'
                 if result[0].boxes.cls[0].cpu().numpy()==4:
-                        label='person'
+                        lbel='person'
                 if result[0].boxes.cls[0].cpu().numpy()==5:
-                        label='train'
+                        lbel='train'
                 if result[0].boxes.cls[0].cpu().numpy()==6:
-                        label='truck'
-                ann.box_label(box=box, label=label, color=(0,128,0))
+                        lbel='truck'
+                ann.box_label(box=box, label=lbel, color=(0,128,0))
             vis=ann.result()
         else:
             vis = image