update config

bmd1905 · Dec 3, 2022 · 54e8a77 · 54e8a77
1 parent 04a8f34
commit 54e8a77
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 13 deletions.
diff --git a/config/vgg-seq2seq.yml b/config/vgg-seq2seq.yml
@@ -1,10 +1,8 @@
-pretrain: 
-    id_or_url: 1nTKlEog9YFK74kPyX0qLwCWi60_YHHk4
-    md5: efcabaa6d3adfca8e52bda2fd7d2ee04
-    cached: /tmp/tranformerorc.pth
+# for train
+pretrain: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
 
-# url or local path
-weights: https://drive.google.com/uc?id=1nTKlEog9YFK74kPyX0qLwCWi60_YHHk4
+# url or local path (for predict)
+weights: https://vocr.vn/data/vietocr/vgg_seq2seq.pth
 
 backbone: vgg19_bn
 cnn:

diff --git a/config/vgg-transformer.yml b/config/vgg-transformer.yml
@@ -1,10 +1,8 @@
-pretrain: 
-    id_or_url: 13327Y1tz1ohsm5YZMyXVMPIOjoOA0OaA
-    md5: af6b46e9295eb1c27ca90bddf8c8729a
-    cached: /tmp/tranformerorc.pth
+# for training
+pretrain: https://vocr.vn/data/vietocr/vgg_transformer.pth
 
-# url or local path
-weights: https://drive.google.com/uc?id=13327Y1tz1ohsm5YZMyXVMPIOjoOA0OaA
+# url or local path (predict)
+weights: https://vocr.vn/data/vietocr/vgg_transformer.pth
 
 backbone: vgg19_bn
 cnn:

diff --git a/setup.py b/setup.py
@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="vietocr",
-    version="0.3.9",
+    version="0.3.10",
     author="pbcquoc",
     author_email="[email protected]",
     description="Transformer base text detection",