From 2d433dcf6fe6df5b636e804c0ac616b420a729ac Mon Sep 17 00:00:00 2001
From: Olivier Dulcy <106678676+odulcy-mindee@users.noreply.github.com>
Date: Mon, 18 Sep 2023 10:32:19 +0200
Subject: [PATCH] feat: :sparkles: PT ViTSTR Small Checkpoint (#1319)

---
 doctr/models/recognition/vitstr/pytorch.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/doctr/models/recognition/vitstr/pytorch.py b/doctr/models/recognition/vitstr/pytorch.py
index 4262ecad22..5f68800125 100644
--- 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),