Merge pull request #22 from Nempickaxe/patch-1

don't download if model file if provided

Author: bedapudi6788

fastpunct/fastpunct.py

@@ -148,11 +148,15 @@ def __init__(self, lang_code="en", weights_path=None, params_path=None):
 
         home = os.path.expanduser("~")
         lang_path = os.path.join(home, '.fastPunct_' + lang_code)
-        weights_path = os.path.join(lang_path, 'checkpoint.h5')
-        params_path = os.path.join(lang_path, 'params.pkl')
-
-        if not os.path.exists(lang_path):
-            os.mkdir(lang_path)
+        if weights_path is None:
+            weights_path = os.path.join(lang_path, 'checkpoint.h5')
+        if params_path is None:
+            params_path = os.path.join(lang_path, 'params.pkl')
+        
+        #if either of the paths are not mentioned, then, make lang directory from home
+        if (params_path is None) or (weights_path is None):
+            if not os.path.exists(lang_path):
+                os.mkdir(lang_path)
 
         if not os.path.exists(weights_path):
             print('Downloading checkpoint', model_links[lang_code]['checkpoint'], 'to', weights_path)