Check the existence of model file during loading.

README.md

@@ -80,6 +80,8 @@ Example paraphrase datasets for Russian (adapted from http://paraphraser.ru/):
 - https://rusvectores.org/static/testsets/paraphrases.tsv.gz
 - https://rusvectores.org/static/testsets/paraphrases_lemm.tsv.gz (lemmatized)
 
-
-The library requires Python >= 3.6.
+# Training your own ELMo
+Currently we provide ELMo training code (updated and improved in the same way compared to the original implementation)
+in a [separate repository](https://github.com/ltgoslo/simple_elmo_training).
+It will be integrated into the `simple_elmo` package in the nearest future.
 

setup.py

@@ -5,7 +5,7 @@
 
 setuptools.setup(
     name="simple_elmo",
-    version="0.3.4",
+    version="0.3.6",
     author="Andrey Kutuzov",
     author_email="[email protected]",
     description="Handy library to work with pre-trained ELMo embeddings in TensorFlow",

simple_elmo/elmo_helpers.py

@@ -44,6 +44,8 @@ def load(self, directory, top=False, max_batch_size=96):
         :param max_batch_size: the maximum allowable batch size during inference
         :return: ELMo batcher, character id placeholders, op object
         """
+        if not os.path.exists(directory):
+            raise SystemExit("Error: model not found!")
         self.batch_size = max_batch_size
         if os.path.isfile(directory) and directory.endswith(".zip"):
             message = """