Brasd99 commited on
Commit
5202de8
1 Parent(s): 60761c3

Fixed prints

Browse files
Files changed (1) hide show
  1. app.py +6 -6
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
- text = st.text_input('Enter some text')
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 text:
69
- st.error('Please write text')
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
- text,
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
  )