diff --git a/ai_models/remote.py b/ai_models/remote.py index 1e54c5c..1f468f5 100644 --- a/ai_models/remote.py +++ b/ai_models/remote.py @@ -21,7 +21,7 @@ def __init__(self, **kwargs): self.cfg["download_assets"] = False self.cfg["assets_extra_dir"] = None self._param = {} - self.api = RemoteClient() + self.api = RemoteAPI() super().__init__(**self.cfg) @@ -88,7 +88,7 @@ def __call__(self, r): return r -class RemoteClient: +class RemoteAPI: def __init__( self, input_file: str = None,