diff --git a/animegan.py b/animegan.py index da3e055..0982a80 100644 --- a/animegan.py +++ b/animegan.py @@ -20,8 +20,8 @@ class Animegan(Operator): super().__init__() if framework == 'pytorch': import importlib.util - path = os.path.join(str(Path("__file__").parent), 'pytorch', 'model.py') - opname = os.path.basename(str(Path("__file__"))).split('.')[0] + path = os.path.join(str(Path(__file__).parent), 'pytorch', 'model.py') + opname = os.path.basename(str(Path(__file__))).split('.')[0] spec = importlib.util.spec_from_file_location(opname, path) module = importlib.util.module_from_spec(spec) spec.loader.exec_module(module)