Spaces:
Running
Running
Fixed prints
Browse files
app.py
CHANGED
@@ -32,8 +32,6 @@ def get_iso_languages():
|
|
32 |
iso_code, language_name = parts
|
33 |
if contains_only_ascii(language_name):
|
34 |
iso_language_dict[language_name] = iso_code
|
35 |
-
else:
|
36 |
-
print(f'Invalid language: {language_name}')
|
37 |
|
38 |
return iso_language_dict
|
39 |
|
@@ -59,14 +57,14 @@ st.image(APP_LOGO, use_column_width=True)
|
|
59 |
st.markdown(APP_DESCRIPTION)
|
60 |
|
61 |
language = st.selectbox('Select a language', languages)
|
62 |
-
|
63 |
input_wav = st.file_uploader("Upload a WAV file", type=["wav"])
|
64 |
|
65 |
if input_wav:
|
66 |
if not input_wav or input_wav is None:
|
67 |
st.error('Please upload wav input audio')
|
68 |
-
elif not
|
69 |
-
st.error('Please write
|
70 |
else:
|
71 |
progress_bar = st.progress(0)
|
72 |
status_text = st.empty()
|
@@ -75,12 +73,14 @@ if input_wav:
|
|
75 |
|
76 |
iso_code = iso_languages[language]
|
77 |
|
|
|
|
|
78 |
update_progress(0, 'Loading TTS model...')
|
79 |
api = TTS(f"tts_models/{iso_code}/fairseq/vits")
|
80 |
|
81 |
update_progress(50, 'Generating audio...')
|
82 |
api.tts_with_vc_to_file(
|
83 |
-
|
84 |
speaker_wav=temp_file.name,
|
85 |
file_path=OUTPUT_FILENAME
|
86 |
)
|
|
|
32 |
iso_code, language_name = parts
|
33 |
if contains_only_ascii(language_name):
|
34 |
iso_language_dict[language_name] = iso_code
|
|
|
|
|
35 |
|
36 |
return iso_language_dict
|
37 |
|
|
|
57 |
st.markdown(APP_DESCRIPTION)
|
58 |
|
59 |
language = st.selectbox('Select a language', languages)
|
60 |
+
prompt = st.text_input('Enter your prompt')
|
61 |
input_wav = st.file_uploader("Upload a WAV file", type=["wav"])
|
62 |
|
63 |
if input_wav:
|
64 |
if not input_wav or input_wav is None:
|
65 |
st.error('Please upload wav input audio')
|
66 |
+
elif not prompt:
|
67 |
+
st.error('Please write prompt')
|
68 |
else:
|
69 |
progress_bar = st.progress(0)
|
70 |
status_text = st.empty()
|
|
|
73 |
|
74 |
iso_code = iso_languages[language]
|
75 |
|
76 |
+
print(f'Language: {language}, prompt: {prompt}')
|
77 |
+
|
78 |
update_progress(0, 'Loading TTS model...')
|
79 |
api = TTS(f"tts_models/{iso_code}/fairseq/vits")
|
80 |
|
81 |
update_progress(50, 'Generating audio...')
|
82 |
api.tts_with_vc_to_file(
|
83 |
+
prompt,
|
84 |
speaker_wav=temp_file.name,
|
85 |
file_path=OUTPUT_FILENAME
|
86 |
)
|