metadata
license: apache-2.0
datasets:
- dataautogpt3/Dalle3
- scrapegraphai/AQL-v1-QA
language:
- en
metrics:
- accuracy
base_model:
- microsoft/resnet-50
new_version: microsoft/resnet-50
pipeline_tag: image-classification
def load_model(model_path, num_classes):
model = create_model(num_classes)
model.load_state_dict(torch.load(model_path))
model.eval()
return model