Update app.py

app.py

@@ -45,6 +45,7 @@ def process_img(image):
                         lbel='train'
                 if result[0].boxes.cls[0].cpu().numpy()==6:
                         lbel='truck'
+                    
                 ann.box_label(box=box, label=lbel, color=(0,128,0))
             vis=ann.result()
         else: