diff --git a/model_server/base/models.py b/model_server/base/models.py index c24ed92d6a2b3723020c4c60c45d6d5041399435..ffaa346ff61f040af5b7b26182af6a09d3d5f6ce 100644 --- a/model_server/base/models.py +++ b/model_server/base/models.py @@ -48,6 +48,11 @@ class Model(ABC): def reload(self): self.load() + @property + def name(self): + return f'{self.__class__}' + + class ImageToImageModel(Model): """ diff --git a/model_server/base/session.py b/model_server/base/session.py index 0e624ab180977a7bd070b29e8b656881d7f1f850..8358a957e329efa06d6509f18c093dc03df55639 100644 --- a/model_server/base/session.py +++ b/model_server/base/session.py @@ -240,8 +240,7 @@ class _Session(object): if key is None: def mid(i): - # TODO: give model the option to report its own name - return f'{ModelClass.__name__}_{i:02d}' + return f'{mi.name}_{i:02d}' while mid(ii) in self.models.keys(): ii += 1