AlienKevin commited on
Commit
758aaaa
1 Parent(s): 64843d8

Change mps device to cpu

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -8,7 +8,7 @@ feature_extractor = WhisperFeatureExtractor.from_pretrained(f"alvanlii/whisper-s
8
  feature_extractor.chunk_length = 3
9
 
10
  # Instantiate the model
11
- device = torch.device("mps")
12
  model = WhisperAudioClassifier().to(device)
13
 
14
  # Load the state dict
 
8
  feature_extractor.chunk_length = 3
9
 
10
  # Instantiate the model
11
+ device = torch.device("cpu")
12
  model = WhisperAudioClassifier().to(device)
13
 
14
  # Load the state dict