Update app.py

app.py

@@ -26,8 +26,9 @@ def apply_nms(orig_prediction, iou_thresh=0.3):
 
 # Draw the bounding box
 def plot_img_bbox(img, target):
+        h,w,c = img.shape
         for box in (target['boxes']):
-            xmin, ymin, xmax, ymax  = int(box[0].cpu()), int(box[1].cpu()), int(box[2].cpu()),int(box[3].cpu())
+            xmin, ymin, xmax, ymax  = int((box[0].cpu()/1024)*w), int((box[1].cpu()/1024)*h), int((box[2].cpu()/1024)*w),int((box[3].cpu()/1024)*h)
             cv2.rectangle(img, (xmin, ymin), (xmax, ymax), (0, 0, 255), 2)
             label = "palm"
             # Add the label and confidence score