safeload

bmd1905 · Dec 9, 2020 · baa7f46 · baa7f46
1 parent 9baddaa
commit baa7f46
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/vietocr/tool/utils.py b/vietocr/tool/utils.py
@@ -17,7 +17,7 @@ def download_weights(id_or_url, cached=None, md5=None, quiet=False):
 def download_config(id):
     url = 'https://raw.githubusercontent.com/pbcquoc/vietocr/master/config/{}'.format(id)
     r = requests.get(url)
-    config = yaml.sale_load(r.text)
+    config = yaml.safe_load(r.text)
 
     return config