Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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=
|
28 |
-
tokenizer = AutoTokenizer.from_pretrained("Norphel/Dz_en", token=
|
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):
|