Update models.py

models.py

@@ -7,7 +7,6 @@ from PIL import Image
 class LinearClassifier(torch.nn.Module):
     def __init__(self, input_dim, output_dim):
         super(LinearClassifier, self).__init__()
-        num_classes = len(index_to_species.keys())
 
         self.linear_head = nn.Sequential(
             nn.Linear(input_dim, 512, bias=True),