Merge pull request #17 from torokati44/master

Fix model loading

Author: minimaxir

textgenrnn/textgenrnn.py

@@ -225,7 +225,7 @@ def save(self, weights_path="textgenrnn_weights_saved.hdf5"):
         self.model.save_weights(weights_path)
 
     def load(self, weights_path):
-        self.model = textgenrnn_model(weights_path, self.num_classes)
+        self.model = textgenrnn_model(self.num_classes, cfg=self.config, weights_path=weights_path)
 
     def reset(self):
         self.config = self.default_config.copy()