[Other] Add a interface to get all pretrained models available from hub model server (#1022)

add get model list
This commit is contained in:
chenjian
2023-01-03 09:45:42 +08:00
committed by GitHub
parent 971cc051f4
commit 42f2e8d22b
3 changed files with 41 additions and 1 deletions

View File

@@ -98,6 +98,20 @@ class ModelServer(object):
except requests.exceptions.ConnectionError as e:
raise ServerConnectionError(self._url)
def get_model_list(self):
'''
Get all pre-trained models information in dataset.
Return:
result(dict): key is category name, value is a list which contains models \
information such as name, format and version.
'''
api = '{}/{}'.format(self._url, 'fastdeploy_listmodels')
try:
result = requests.get(api, timeout=self._timeout)
return result.json()
except requests.exceptions.ConnectionError as e:
raise ServerConnectionError(self._url)
def is_connected(self):
return self.check(self._url)