--- 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 --- ```python def load_model(model_path, num_classes): model = create_model(num_classes) model.load_state_dict(torch.load(model_path)) model.eval() return model