diff --git a/Generate.py b/Generate.py index 9acd4356..dacaeb75 100644 --- a/Generate.py +++ b/Generate.py @@ -216,7 +216,7 @@ def main(args=None, callback=ERmain): def read_weights_yaml(path): try: - if urllib.parse.urlparse(path).scheme: + if urllib.parse.urlparse(path).scheme in ('https', 'file'): yaml = str(urllib.request.urlopen(path).read(), "utf-8") else: with open(path, 'rb') as f: