Update app.py

app.py

@@ -3,7 +3,7 @@ import torch
 import torch.nn as nn
 from torchvision import transforms, models
 import pickle
-
+from resnest.torch import resnest50
 
 with open('class_names.pkl', 'rb') as f:
     class_names = pickle.load(f)
@@ -11,7 +11,7 @@ with open('class_names.pkl', 'rb') as f:
 # 加载训练好的模型
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
-model = models.resnet50(weights=None)
+model = resnest50(pretrained=None)
 model.fc = nn.Sequential(
     nn.Dropout(0.2),
     nn.Linear(model.fc.in_features, len(class_names))