anzorq commited on
Commit
1c803c5
·
verified ·
1 Parent(s): b5d1a71

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -3,8 +3,8 @@ import torch
3
  import torchaudio
4
  from transformers import AutoModelForCTC, Wav2Vec2BertProcessor
5
 
6
- model = AutoModelForCTC.from_pretrained("anzorq/output")
7
- processor = Wav2Vec2BertProcessor.from_pretrained("anzorq/output")
8
 
9
  def transcribe_speech(audio):
10
  # Load the audio file
 
3
  import torchaudio
4
  from transformers import AutoModelForCTC, Wav2Vec2BertProcessor
5
 
6
+ model = AutoModelForCTC.from_pretrained("anzorq/w2v-bert-2.0-kbd")
7
+ processor = Wav2Vec2BertProcessor.from_pretrained("anzorq/w2v-bert-2.0-kbd")
8
 
9
  def transcribe_speech(audio):
10
  # Load the audio file