download function in python module #5

Filter-Bubble · Jan 11, 2021 · c58aaf3 · c58aaf3
1 parent 6701dd4
commit c58aaf3
Show file tree

Hide file tree

Showing 2 changed files with 16 additions and 12 deletions.
diff --git a/setup.sh b/setup.sh
diff --git a/stroll/util.py b/stroll/util.py
@@ -0,0 +1,16 @@
+from pathlib import Path
+import urllib.request
+
+
+def download_srl_model(datapath):
+    datapath = Path(datapath)
+    fname_fasttext = datapath / 'fasttext.model.bin'
+    fname_model = datapath / 'srl.pt'
+    if not fname_fasttext.exists():
+        url = 'https://surfdrive.surf.nl/files/index.php/s/085yxFcRmn0osMw/download'
+        urllib.request.urlretrieve(url, fname_fasttext)
+
+    if not fname_model.exists():
+        url = 'https://surfdrive.surf.nl/files/index.php/s/kOgUm0oEpmx5HiZ/download'
+        urllib.request.urlretrieve(url, fname_model)
+    return fname_fasttext, fname_model