Skip to content

Commit

Permalink
change column Name to Model
Browse files Browse the repository at this point in the history
  • Loading branch information
cxnt committed May 23, 2024
1 parent 590bde3 commit 8c6871b
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion serve/src/main.py
Original file line number Diff line number Diff line change
Expand Up @@ -295,7 +295,7 @@ def get_models(self):
model_config[model_meta["model_name"]]["config_url"] = os.path.dirname(model_yml_url)
for model in model_info["Models"]:
checkpoint_info = OrderedDict()
checkpoint_info["Name"] = model["Name"]
checkpoint_info["Model"] = model["Name"]
checkpoint_info["Backbone"] = model["Metadata"]["backbone"]
checkpoint_info["Method"] = model["In Collection"]
checkpoint_info["Dataset"] = model["Results"][0]["Dataset"]
Expand Down

0 comments on commit 8c6871b

Please sign in to comment.