raihanrifaldi commited on
Commit
6cfc374
·
1 Parent(s): 718aed2

update source_languages

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -29,14 +29,14 @@ import psutil
29
 
30
  whisper_models = ["tiny", "base", "small", "medium", "large-v1", "large-v2"]
31
  source_languages = {
32
- "English": "English",
33
  # "zh": "Chinese",
34
  # "de": "German",
35
  # "es": "Spanish",
36
  # "ru": "Russian",
37
  # "ko": "Korean",
38
  # "fr": "French",
39
- "Japan": "Japanese",
40
  # "pt": "Portuguese",
41
  # "tr": "Turkish",
42
  # "pl": "Polish",
@@ -45,7 +45,7 @@ source_languages = {
45
  # "ar": "Arabic",
46
  # "sv": "Swedish",
47
  # "it": "Italian",
48
- "Indonesia": "Indonesian"
49
  # "hi": "Hindi",
50
  # "fi": "Finnish",
51
  # "vi": "Vietnamese",
@@ -351,7 +351,7 @@ video_in = gr.Video(label="Video file", mirror_webcam=False)
351
  youtube_url_in = gr.Textbox(label="Youtube url", lines=1, interactive=True)
352
  df_init = pd.DataFrame(columns=['Start', 'End', 'Speaker', 'Text'])
353
  memory = psutil.virtual_memory()
354
- selected_source_lang = gr.Dropdown(choices=source_language_list, type="value", value="Indonesia", label="Spoken language in video", interactive=True)
355
  selected_whisper_model = gr.Dropdown(choices=whisper_models, type="value", value="base", label="Selected Whisper model", interactive=True)
356
  number_speakers = gr.Number(precision=0, value=0, label="Input number of speakers for better results. If value=0, model will automatic find the best number of speakers", interactive=True)
357
  system_info = gr.Markdown(f"*Memory: {memory.total / (1024 * 1024 * 1024):.2f}GB, used: {memory.percent}%, available: {memory.available / (1024 * 1024 * 1024):.2f}GB*")
 
29
 
30
  whisper_models = ["tiny", "base", "small", "medium", "large-v1", "large-v2"]
31
  source_languages = {
32
+ "English": "en",
33
  # "zh": "Chinese",
34
  # "de": "German",
35
  # "es": "Spanish",
36
  # "ru": "Russian",
37
  # "ko": "Korean",
38
  # "fr": "French",
39
+ # "ja": "Japanese",
40
  # "pt": "Portuguese",
41
  # "tr": "Turkish",
42
  # "pl": "Polish",
 
45
  # "ar": "Arabic",
46
  # "sv": "Swedish",
47
  # "it": "Italian",
48
+ "Indonesian": "id",
49
  # "hi": "Hindi",
50
  # "fi": "Finnish",
51
  # "vi": "Vietnamese",
 
351
  youtube_url_in = gr.Textbox(label="Youtube url", lines=1, interactive=True)
352
  df_init = pd.DataFrame(columns=['Start', 'End', 'Speaker', 'Text'])
353
  memory = psutil.virtual_memory()
354
+ selected_source_lang = gr.Dropdown(choices=source_language_list, type="value", value="en", label="Spoken language in video", interactive=True)
355
  selected_whisper_model = gr.Dropdown(choices=whisper_models, type="value", value="base", label="Selected Whisper model", interactive=True)
356
  number_speakers = gr.Number(precision=0, value=0, label="Input number of speakers for better results. If value=0, model will automatic find the best number of speakers", interactive=True)
357
  system_info = gr.Markdown(f"*Memory: {memory.total / (1024 * 1024 * 1024):.2f}GB, used: {memory.percent}%, available: {memory.available / (1024 * 1024 * 1024):.2f}GB*")