Norphel commited on
Commit
7f18afd
·
verified ·
1 Parent(s): 5d18df1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -4
app.py CHANGED
@@ -7,6 +7,9 @@ from transformers import Wav2Vec2ForCTC, Wav2Vec2Processor
7
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
8
  from transformers import VitsModel, AutoTokenizer
9
  import tempfile
 
 
 
10
 
11
  st.title("Dzongkha Speech-to-Text")
12
 
@@ -24,14 +27,14 @@ def load_asr_model():
24
 
25
  @st.cache_resource
26
  def load_translation_model():
27
- model = AutoModelForSeq2SeqLM.from_pretrained("Norphel/Dz_en", token="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")
28
- tokenizer = AutoTokenizer.from_pretrained("Norphel/Dz_en", token="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")
29
  return model, tokenizer
30
 
31
  @st.cache_resource
32
  def load_tts_model():
33
- model = VitsModel.from_pretrained("Norphel/MMS-TTS-Dzo-N3")
34
- tokenizer = AutoTokenizer.from_pretrained("Norphel/MMS-TTS-Dzo-N3")
35
  return model, tokenizer
36
 
37
  def generate_voice(text):
 
7
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM, pipeline
8
  from transformers import VitsModel, AutoTokenizer
9
  import tempfile
10
+ import os
11
+
12
+ api_key = os.getenv("hf_token")
13
 
14
  st.title("Dzongkha Speech-to-Text")
15
 
 
27
 
28
  @st.cache_resource
29
  def load_translation_model():
30
+ model = AutoModelForSeq2SeqLM.from_pretrained("Norphel/Dz_en", token=api_key)
31
+ tokenizer = AutoTokenizer.from_pretrained("Norphel/Dz_en", token=api_key)
32
  return model, tokenizer
33
 
34
  @st.cache_resource
35
  def load_tts_model():
36
+ model = VitsModel.from_pretrained("Norphel/MMS-TTS-Dzo-N3", token = api_key)
37
+ tokenizer = AutoTokenizer.from_pretrained("Norphel/MMS-TTS-Dzo-N3", token = api_key)
38
  return model, tokenizer
39
 
40
  def generate_voice(text):