Update app.py

app.py

@@ -214,13 +214,13 @@ def do_prediction(model_name, img):
                 if x <= 30:
                     w += x
                     x = 0
-                if (img.shape[1] - (x + w)) <= 30:
-                    w = w + (img.shape[1] - (x + w))
+                if (img_org.shape[1] - (x + w)) <= 30:
+                    w = w + (img_org.shape[1] - (x + w))
                 if y <= 30:
                     h = h + y
                     y = 0
-                if (img.shape[0] - (y + h)) <= 30:
-                    h = h + (img.shape[0] - (y + h))
+                if (img_org.shape[0] - (y + h)) <= 30:
+                    h = h + (img_org.shape[0] - (y + h))
 
                 box = [x, y, w, h]