Update app.py

app.py

@@ -53,13 +53,14 @@ def preprocess_image(image):
     #img0 = cv2.imread(image_path)
     img0 = letterbox(image, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
     img = letterbox(img0, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
-    print("in preprocess:",img.shape)
-    img = img.transpose(2, 0, 1)[::-1]  # Convert BGR to RGB, to 3x416x416
+    #print("in preprocess:",img.shape)
+    #img = img.transpose(2, 0, 1)[::-1]  # Convert BGR to RGB, to 3x416x416
     img = np.ascontiguousarray(img)
     img = torch.from_numpy(img).to(device)
     img = img.float()  # uint8 to fp16/32
     img /= 255.0  # 0 - 255 to 0.0 - 1.0
     if img.ndimension() == 3:
+        img = img.transpose(2, 0, 1)[::-1]  # Convert BGR to RGB, 
         img = img.unsqueeze(0)
     return img, img0