From bcd521463b9631c1fbf12b1aa0d123e5f1a5a7d7 Mon Sep 17 00:00:00 2001
From: Olivier Dulcy <106678676+odulcy-mindee@users.noreply.github.com>
Date: Tue, 23 Apr 2024 12:03:09 +0200
Subject: [PATCH] feat: :sparkles: torch crop orientation chkpt + rename other
 model (#1563)

---
 doctr/models/classification/mobilenet/pytorch.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/doctr/models/classification/mobilenet/pytorch.py b/doctr/models/classification/mobilenet/pytorch.py
index 2f040431b0..a8b66e1a1d 100644
--- a/doctr/models/classification/mobilenet/pytorch.py
+++ b/doctr/models/classification/mobilenet/pytorch.py
@@ -56,14 +56,14 @@
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 256, 256),
         "classes": [0, -90, 180, 90],
-        "url": "https://doctr-static.mindee.com/models?id=v0.8.1/mobilenet_v3_small_orientation-97a3028f.pt&src=0",
+        "url": "https://doctr-static.mindee.com/models?id=v0.8.1/mobilenet_v3_small_crop_orientation-97a3028f.pt&src=0",
     },
     "mobilenet_v3_small_page_orientation": {
         "mean": (0.694, 0.695, 0.693),
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 512, 512),
         "classes": [0, -90, 180, 90],
-        "url": None,
+        "url": "https://doctr-static.mindee.com/models?id=v0.8.1/mobilenet_v3_small_page_orientation-1409045c.pt&src=0",
     },
 }