Update app.py

app.py

@@ -119,7 +119,7 @@ def detect(img,thr=0.2,trained_dataset='aitod'):
           for idx,b in enumerate(box):
               label_i = lab[idx].item()
               draw.rectangle(list(b), outline=label_color_dict[label_i%8+1], )
-              draw.text((b[0], b[1]), text=coco_dict[label_i], fill='blue', )
+              draw.text((b[0], b[1]), text=coco_dict[label_i+1], fill='blue', )
   
   #save_path = Path('./output') / img_path.name
   return img