sidpanda commited on
Commit
164f25e
1 Parent(s): 6011925

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -16,10 +16,10 @@ import librosa
16
  import pyloudnorm as pyln
17
 
18
  # Load the models and tokenizers
19
- # model1 = Wav2Vec2ForCTC.from_pretrained("ai4bharat/indicwav2vec-hindi")
20
- # tokenizer1 = Wav2Vec2Tokenizer.from_pretrained("ai4bharat/indicwav2vec-hindi")
21
- model1 = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-large-960h")
22
- tokenizer1 = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-large-960h")
23
  # Loading the tokenizer and model from Hugging Face's model hub.
24
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
25
  model = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=os.environ.get('HF_TOKEN'))
 
16
  import pyloudnorm as pyln
17
 
18
  # Load the models and tokenizers
19
+ model1 = Wav2Vec2ForCTC.from_pretrained("ai4bharat/indicwav2vec-hindi")
20
+ tokenizer1 = Wav2Vec2Tokenizer.from_pretrained("ai4bharat/indicwav2vec-hindi")
21
+ # model1 = Wav2Vec2ForCTC.from_pretrained("facebook/wav2vec2-large-960h")
22
+ # tokenizer1 = Wav2Vec2Tokenizer.from_pretrained("facebook/wav2vec2-large-960h")
23
  # Loading the tokenizer and model from Hugging Face's model hub.
24
  tokenizer = AutoTokenizer.from_pretrained("google/gemma-2b")
25
  model = AutoModelForCausalLM.from_pretrained("google/gemma-2b", token=os.environ.get('HF_TOKEN'))