Update app.py

app.py

@@ -4,6 +4,7 @@ import torch.nn as nn
 from torchvision import models, transforms
 from PIL import Image
 import numpy as np
+from torchvision.models import ResNet50_Weights
 
 # -----------------------------
 # Device
@@ -17,7 +18,7 @@ print(f"Using device: {device}")
 class FoodIngredientClassifier(nn.Module):
     def __init__(self, num_classes):
         super().__init__()
-        self.backbone = models.resnet50(weights=None)
+        self.backbone = models.resnet50(weights=models.ResNet50_Weights.IMAGENET1K_V1)
         num_features = self.backbone.fc.in_features
         self.backbone.fc = nn.Sequential(
             nn.Dropout(0.5),