Spaces:

spycoder
/

wav2vec

Runtime error

spycoder commited on Jun 6, 2023

Commit

e1cd816

•

1 Parent(s): 580a2d6

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -12,7 +12,7 @@ import torch.nn.functional as F
 from torch.utils.data import Dataset, DataLoader
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 from collections import Counter
-processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h"
 model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
 model_path = "dysarthria_classifier12.pth"
 if os.path.exists(model_path):

 from torch.utils.data import Dataset, DataLoader
 from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor, Wav2Vec2ForSequenceClassification
 from collections import Counter
+processor = Wav2Vec2Processor.from_pretrained("facebook/wav2vec2-base-960h")
 model = Wav2Vec2ForSequenceClassification.from_pretrained("facebook/wav2vec2-base-960h", num_labels=2).to(device)
 model_path = "dysarthria_classifier12.pth"
 if os.path.exists(model_path):