Update app.py

app.py

@@ -84,7 +84,10 @@ def preprocess_image(img: Image.Image):
     img_np = ensure_rgb(np.array(img))
     img_resized = cv2.resize(img_np, (IMG_SIZE, IMG_SIZE))
     img_norm = img_resized.astype(np.float32) / 255.0
-    img_norm = (img_norm - 0.5) / 0.5
+    mean = np.array([0.485, 0.456, 0.406])
+    std  = np.array([0.229, 0.224, 0.225])
+    img_norm = (img_norm - mean) / std
+
     tensor = torch.from_numpy(img_norm).permute(2, 0, 1).unsqueeze(0)
     return tensor.to(DEVICE), img_np