fix load_model to use absolute path

9 years ago · 5248437d06
parent d1ab2b1fa3
commit 5248437d06
1 changed files with 1 additions and 1 deletions
--- a/enhance.py
+++ b/enhance.py
@ -387,7 +387,7 @@ class Model(object):
            error("Model file with pre-trained convolution layers not found. Download it here...",
                  "https://github.com/alexjc/neural-enhance/releases/download/v%s/%s"%(__version__, self.get_filename()))
        print('  - Loaded file `{}` with trained model.'.format(self.get_filename()))
-        return pickle.load(bz2.open(self.get_filename(), 'rb'))
+        return pickle.load(bz2.open(self.get_filename(absolute=True), 'rb'))

    def load_generator(self, params):
        if len(params) == 0: return