Update app.py

app.py

@@ -19,7 +19,7 @@ def load_model():
     in_features = model.classifier[2].in_features
     model.classifier[2] = nn.Sequential(
         nn.Dropout(p=0.5), # Standard dropout layer (index 0)
-        nn.Linear(in_features, len(class_names)) # Linear layer (index 1)
+        nn.Linear(in_features, 102) # Linear layer (index 1)
     )
     
     state_dict = torch.load("convnext_caltech101.pth", map_location=device)