Mirrors/modelscope
1
0
Fork 0
mirror of https://github.com/modelscope/modelscope.git synced 2025-12-16 16:27:45 +01:00
Code Issues Packages Projects Releases Wiki Activity

fix: check tasks

Browse Source
This commit is contained in:
suluyan
2025-12-09 21:45:52 +08:00
parent 4be91830df
commit d351698f28
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
modelscope/hub/api.py
View File

@@ -2908,6 +2908,15 @@ class HubApi:
if repo_type == REPO_TYPE_MODEL:
model_info = self.get_model(model_id=repo_id)
path = f'{self.endpoint}/api/v1/models/{repo_id}'
tasks = model_info.get('Tasks')
if isinstance(tasks, list) and tasks:
first = tasks[0]
if isinstance(first, dict) and first:
model_tasks = first.get('name')
else:
model_tasks = ''
else:
model_tasks = ''
payload = {
'ChineseName': model_info.get('ChineseName', ''),
'ModelFramework': model_info.get('ModelFramework', 'Pytorch'),
@@ -2921,7 +2930,7 @@ class HubApi:
'SubScientificField': model_info.get('SubScientificField', None),
'ScientificField': model_info.get('NEXA', {}).get('ScientificField', ''),
'Source': model_info.get('NEXA', {}).get('Source', ''),
'ModelTask': model_info.get('Tasks', [{}])[0].get('name'),
'ModelTask': model_tasks,
'License': model_info.get('License', ''),
}
elif repo_type == REPO_TYPE_DATASET: