Update app.py
Browse files
app.py
CHANGED
@@ -98,22 +98,20 @@ class App:
|
|
98 |
nllb_params = translation_params["nllb"]
|
99 |
|
100 |
with gr.Row():
|
101 |
-
|
102 |
-
|
103 |
-
|
104 |
-
|
105 |
-
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
|
113 |
-
|
114 |
-
|
115 |
-
cb_translate_output = gr.Checkbox(value=translation_params["translate_output"], label="Translate output",interactive=True)
|
116 |
-
|
117 |
# with gr.Accordion("Speaker diarization", open=False, visible=True):
|
118 |
# cb_diarize = gr.Checkbox(value=diarization_params["is_diarize"], label="Use diarization",interactive=True)
|
119 |
# tb_hf_token = gr.Text(label="Token", value=diarization_params["hf_token"],info="Required to use diarization")
|
|
|
98 |
nllb_params = translation_params["nllb"]
|
99 |
|
100 |
with gr.Row():
|
101 |
+
with gr.Column(scale=2):
|
102 |
+
dd_model = gr.Dropdown(choices=self.whisper_inf.available_models, value=whisper_params["model_size"],label="Model", info="Larger models will increase the quality of the transcription, but reduce performance")
|
103 |
+
dd_lang = gr.Dropdown(choices=["Automatic Detection"] + self.whisper_inf.available_langs,value=whisper_params["lang"], label="Language", info="If the language is known upfront, always set it manually")
|
104 |
+
cb_translate = gr.Checkbox(value=whisper_params["is_translate"], label="Translate to English",interactive=True, visible=False)
|
105 |
+
#dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
|
106 |
+
dd_file_format = gr.Dropdown(choices=["TXT","SRT"], value="TXT", label="Output format", info="Output preview format", visible=False)
|
107 |
+
with gr.Column(scale=2):
|
108 |
+
dd_translate_model = gr.Dropdown(choices=self.nllb_inf.available_models, value=nllb_params["model_size"],label="Model", info="Model used for translation")
|
109 |
+
#dd_target_lang = gr.Dropdown(choices=self.nllb_inf.available_target_langs, value=nllb_params["target_lang"],label="Target Language")
|
110 |
+
dd_target_lang = gr.Dropdown(choices=["English","Dutch","French","German"], value=nllb_params["target_lang"],label="Target language", info="Language used for output translation")
|
111 |
+
with gr.Column(scale=1):
|
112 |
+
cb_timestamp = gr.Checkbox(value=whisper_params["add_timestamp"], label="Add timestamp to output file",interactive=True)
|
113 |
+
cb_translate_output = gr.Checkbox(value=translation_params["translate_output"], label="Translate output",interactive=True)
|
114 |
+
|
|
|
|
|
115 |
# with gr.Accordion("Speaker diarization", open=False, visible=True):
|
116 |
# cb_diarize = gr.Checkbox(value=diarization_params["is_diarize"], label="Use diarization",interactive=True)
|
117 |
# tb_hf_token = gr.Text(label="Token", value=diarization_params["hf_token"],info="Required to use diarization")
|