davidmeikle commited on
Commit
b278503
·
verified ·
1 Parent(s): ac24704

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -139,8 +139,7 @@ class PhonemeTranscriber:
139
  def _initialize_model(self) -> ModelConfig:
140
  model_name = "facebook/wav2vec2-lv-60-espeak-cv-ft"
141
  processor = Wav2Vec2Processor.from_pretrained(model_name)
142
- model = Wav2Vec2ForCTC.from_pretrained(model_name).to(self.device)
143
- model.eval()
144
 
145
  return ModelConfig(
146
  name=model_name,
 
139
  def _initialize_model(self) -> ModelConfig:
140
  model_name = "facebook/wav2vec2-lv-60-espeak-cv-ft"
141
  processor = Wav2Vec2Processor.from_pretrained(model_name)
142
+ model = Wav2Vec2ForCTC.from_pretrained(model_name)
 
143
 
144
  return ModelConfig(
145
  name=model_name,