Update app.py

app.py

@@ -13,9 +13,10 @@ config.num_labels = 8  # Set it to the number of classes in your SER task
 
 # Load the pre-trained model with the modified configuration
 model = HubertForSequenceClassification.from_pretrained("superb/hubert-large-superb-er", config=config, ignore_mismatched_sizes=True)
+model.to(device)
 checkpoint =torch.load(model_name, map_location = device)
 model.load_state_dict(checkpoint)
-model.to(device)
+
 
 model.eval()