feat: ✨ PT fast small checkpoint (mindee#1529)

odulcy-mindee · Mar 26, 2024 · 8c85c36 · 8c85c36
1 parent 8765b2d
commit 8c85c36
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/detection/fast/pytorch.py b/doctr/models/detection/fast/pytorch.py
@@ -32,7 +32,7 @@
         "input_shape": (3, 1024, 1024),
         "mean": (0.798, 0.785, 0.772),
         "std": (0.264, 0.2749, 0.287),
-        "url": None,
+        "url": "https://doctr-static.mindee.com/models?id=v0.8.1/fast_small-10952cc1.pt&src=0",
     },
     "fast_base": {
         "input_shape": (3, 1024, 1024),