Update app.py

app.py

@@ -51,8 +51,8 @@ model.eval()  # Set the model to evaluation mode
 #def preprocess_image(image_path):
 def preprocess_image(image):
     #img0 = cv2.imread(image_path)
-    #img = letterbox(img0, 640, stride=32, auto=True)[0]  # Resize and pad to 640x640
     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
     img = img.transpose(2, 0, 1)[::-1]  # Convert BGR to RGB, to 3x416x416
     img = np.ascontiguousarray(img)
     img = torch.from_numpy(img).to(device)