model.py

import torch
import torchvision
from torchvision.models import EfficientNet_B2_Weights , efficientnet_b2

efficient_weight = EfficientNet_B2_Weights.DEFAULT

efficient_transformer = efficient_weight.transforms()

efficient_model = torch.load("efficient_model_101.pth" , map_location=torch.device('cpu'))