Upload 7704_inference.py

7704_inference.py

@@ -1,13 +1,11 @@
 import json
 import time
 from PIL import Image
-from timm.data import resolve_data_config
 import torch
 from torchvision.transforms import transforms
 
 model = torch.load('model.pth').to("cuda")
 model.eval()
-config = resolve_data_config({}, model=model)
 transform = transforms.Compose([
                 transforms.Resize((224, 224)),
                 transforms.ToTensor(),