vankienemk commited on
Commit
2750922
·
verified ·
1 Parent(s): 36442e1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
5
  from scipy.signal import resample
6
 
7
  # Load model
8
- processor = Wav2Vec2Processor.from_pretrained("nguyenvulebinh/wav2vec2-base-vietnamese-250h")
9
- model = Wav2Vec2ForCTC.from_pretrained("nguyenvulebinh/wav2vec2-base-vietnamese-250h")
10
  model = torch.compile(model)
11
  def transcribe(audio):
12
  if audio is None:
 
5
  from scipy.signal import resample
6
 
7
  # Load model
8
+ processor = Wav2Vec2Processor.from_pretrained("StevenLe456/viet_tones_model")
9
+ model = Wav2Vec2ForCTC.from_pretrained("StevenLe456/viet_tones_model")
10
  model = torch.compile(model)
11
  def transcribe(audio):
12
  if audio is None: