Update app.py

app.py

@@ -19,7 +19,7 @@ num_classes = len(list(index_to_species.keys()))
 
 # Load the model
 classify_model = models.DinoVisionTransformerClassifier(num_classes)
-classify_model.load_state_dict(torch.load("best_dinov2_both_2023-11-21_07-44-35.pth"))
+classify_model.load_state_dict(torch.load("best_dinov2_both_2023-11-21_07-44-35.pth"), map_location=torch.device('cpu'))
 classify_model.eval()