Update app.py

app.py

@@ -24,9 +24,9 @@ num_classes = len(list(index_to_species.keys()))
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
 # Load the model
-linear_model_name = 'linear_2025-07-08.pt'
+linear_model_name = 'linear_model.pt'
 classify_model = models.LinearClassifier(input_dim=768, output_dim=num_classes)
-classify_model.load_state_dict(torch.load(os.path.join('models', linear_model_name)))
+classify_model.load_state_dict(torch.load(linear_model_name))
 
 k = 5