Update app.py

app.py

@@ -29,22 +29,22 @@ def process_img(image):
         lbel=''
         if len(result[0].boxes)>0:
             ann=Annotator(im=image)
-            boxes=result[0].boxes.xyxy
+            boxes=result[0].boxes
             for element in boxes:
-                box=np.array(element.cpu()).flatten()
-                if result[0].boxes.cls[0].cpu().numpy()==2:
+                box=np.array(element.xyxy.cpu()).flatten()
+                if element.cls[0].cpu().numpy()==2.0:
                         lbel='car'
-                if result[0].boxes.cls[0].cpu().numpy()==0:
+                if element.cls[0].cpu().numpy()==0.0:
                         lbel='bicycle'
-                if result[0].boxes.cls[0].cpu().numpy()==1:
+                if element.cls[0].cpu().numpy()==1.0:
                         lbel='bus'
-                if result[0].boxes.cls[0].cpu().numpy()==3:
+                if element.cls[0].cpu().numpy()==3.0:
                         lbel='motorcycle'
-                if result[0].boxes.cls[0].cpu().numpy()==4:
+                if element.cls[0].cpu().numpy()==4.0:
                         lbel='person'
-                if result[0].boxes.cls[0].cpu().numpy()==5:
+                if element.cls[0].cpu().numpy()==5.0:
                         lbel='train'
-                if result[0].boxes.cls[0].cpu().numpy()==6:
+                if element.cls[0].cpu().numpy()==6.0:
                         lbel='truck'
                     
                 ann.box_label(box=box, label=lbel, color=(0,128,0))