spycoder commited on
Commit
df4dfab
·
1 Parent(s): 223eb95

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ from torch.utils.data import Dataset, DataLoader
13
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
14
  from collections import Counter
15
 
16
- device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
17
  processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
18
  model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
19
  model_path = "dysarthria_classifier12.pth"
 
13
  from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
14
  from collections import Counter
15
 
16
+ device = torch.device("cpu")
17
  processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
18
  model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
19
  model_path = "dysarthria_classifier12.pth"