vankienemk commited on
Commit
15eda7f
·
verified ·
1 Parent(s): 2750922

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("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:
 
5
  from scipy.signal import resample
6
 
7
  # Load model
8
+ processor = Wav2Vec2Processor.from_pretrained("Menlo/Ichigo-whisper-v0.1")
9
+ model = Wav2Vec2ForCTC.from_pretrained("Menlo/Ichigo-whisper-v0.1")
10
  model = torch.compile(model)
11
  def transcribe(audio):
12
  if audio is None: