updated model class name

app.py

@@ -4,7 +4,7 @@ import torch.nn.functional as F
 from PIL import Image
 import json
 from torchvision import transforms
-from model import resnet50
+from model import ResNet
 
 # Load class labels from local file
 with open("imagenet_classes.json", "r") as f:
@@ -12,7 +12,7 @@ with open("imagenet_classes.json", "r") as f:
 
 # Load model
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-model = resnet50(num_classes=1000, drop_path_rate=0.0, use_blurpool=True)
+model = ResNet(num_classes=1000, drop_path_rate=0.0, use_blurpool=True)
 model.load_state_dict(torch.load("best_resnet50_imagenet_1k.pt", map_location=device))
 model.to(device)
 model.eval()