Update app.py

app.py

@@ -6,12 +6,12 @@ import torchvision.models as models
 
 # Load model
 model = models.resnet18()
-model.fc = torch.nn.Linear(model.fc.in_features, 3)  # Change '3' to the number of your classes
+model.fc = torch.nn.Linear(model.fc.in_features, 13) # Change '3' to the number of your classes
 model.load_state_dict(torch.load("animal_classifier.pth", map_location=torch.device('cpu')))
 model.eval()
 
 # Class labels (update as needed)
-class_labels = ["cat", "dog", "horse"]  # <-- Replace with your real class names
+class_labels = ["antelope", "badger", "bat", "bear", "bee", "beetle","jellyfish","kangaroo","koala", "ladybug","leopard","lion","lizard"]  # <-- Replace with your real class names
 
 # Transform
 transform = transforms.Compose([