Update app.py

app.py

@@ -4,7 +4,7 @@ import torch
 import soundfile as sf
 
 # Load model and tokenizer
-device = torch.device('cpu')
+device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
 model_name = "model_hubert_finetuned_nopeft.pth"  # Replace with your model path or Hugging Face model hub path
 config = HubertConfig.from_pretrained("superb/hubert-large-superb-er")
 config.id2label = {0: 'neu', 1: 'hap', 2: 'ang', 3: 'sad', 4: 'dis', 5: 'sur', 6: 'fea', 7: 'cal'}