Update app.py

app.py

@@ -22,7 +22,7 @@ class MyCarClassifier(torch.nn.Module):
 # 2. Load model weights
 # --------------------------
 model = MyCarClassifier()
-model.load_state_dict(torch.load("best_stanford_cars_transfer_model.pth", map_location="cpu"))
+model.load_state_dict(torch.load("best_stanford_cars_transfer_model.pth", map_location="cpu"), strict=False)
 model.eval()  # important for inference
 
 # --------------------------