feat: ✨ PT ViTSTR Small Checkpoint (#1319)

mindee · Sep 18, 2023 · 2d433dc · 2d433dc
1 parent 7ab0ece
commit 2d433dc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/recognition/vitstr/pytorch.py b/doctr/models/recognition/vitstr/pytorch.py
@@ -25,7 +25,7 @@
         "std": (0.299, 0.296, 0.301),
         "input_shape": (3, 32, 128),
         "vocab": VOCABS["french"],
-        "url": None,
+        "url": "https://doctr-static.mindee.com/models?id=v0.7.0/vitstr_small-fcd12655.pt&src=0",
     },
     "vitstr_base": {
         "mean": (0.694, 0.695, 0.693),