try to fix mismatch

app.py

@@ -7,14 +7,15 @@ from torchvision import transforms
 # 1. Define your model class
 # --------------------------
 # Make sure this matches the architecture you used to train your model
+
+from torchvision.models import resnet50
+
 class MyCarClassifier(torch.nn.Module):
-    def __init__(self, num_classes=196):  # Stanford Cars has 196 classes
+    def __init__(self, num_classes=196):
         super(MyCarClassifier, self).__init__()
-        # Example: replace with your actual model architecture
-        from torchvision.models import resnet50
         self.model = resnet50(pretrained=False)
-        self.model.fc = torch.nn.Linear(self.model.fc.in_features, num_classes)
-    
+        in_ch = self.model.fc.in_features
+        self.model.fc = torch.nn.Linear(in_ch, num_classes)
     def forward(self, x):
         return self.model(x)