feat: ✨ TF ViTSTR Base checkpoint (mindee#1306)

odulcy-mindee · Sep 8, 2023 · ccfc3e0 · ccfc3e0
1 parent f2d399f
commit ccfc3e0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/doctr/models/recognition/vitstr/tensorflow.py b/doctr/models/recognition/vitstr/tensorflow.py
@@ -30,7 +30,7 @@
         "std": (0.299, 0.296, 0.301),
         "input_shape": (32, 128, 3),
         "vocab": VOCABS["french"],
-        "url": None,
+        "url": "https://doctr-static.mindee.com/models?id=v0.6.0/vitstr_base-2889159a.zip&src=0",
     },
 }