Spaces:
Build error
Build error
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')) | |