Update app.py

app.py

@@ -59,7 +59,7 @@ orientation_labels = {
 
 model_race = RaceClassifier(n_classes=4)
 model_race.to(device)
-model_race.load_state_dict(torch.load('best_model_race.pt', map_location=torch.device('cpu')))
+model_race.load_state_dict(torch.load('best_model_race_last.pt', map_location=torch.device('cpu')))
 
 model_age = RaceClassifier(n_classes=3)
 model_age.to(device)