LAP-DEV commited on
Commit
dd842cb
·
verified ·
1 Parent(s): 7bd1e32

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -29
app.py CHANGED
@@ -102,7 +102,12 @@ class App:
102
  cb_timestamp = gr.Checkbox(value=whisper_params["add_timestamp"], label="Add timestamp to output file",interactive=True, visible=True)
103
  #dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
104
  dd_file_format = gr.Dropdown(choices=["TXT","SRT"], value="TXT", label="Output format", visible=False, info="Output preview format")
105
-
 
 
 
 
 
106
  # with gr.Accordion("Speaker diarization", open=False, visible=True):
107
  # cb_diarize = gr.Checkbox(value=diarization_params["is_diarize"], label="Use diarization",interactive=True)
108
  # tb_hf_token = gr.Text(label="Token", value=diarization_params["hf_token"],info="Required to use diarization")
@@ -319,34 +324,6 @@ class App:
319
  with gr.Column():
320
  gr.Markdown(device_info, label="Hardware info & installed packages")
321
 
322
- with gr.TabItem("Translation"): # tab3
323
- with gr.Row():
324
- file_subs = gr.Files(type="filepath", label="Upload Subtitle Files to translate here",file_types=['.vtt', '.srt'])
325
-
326
- with gr.Row():
327
- dd_model_size = gr.Dropdown(label="Model", value=nllb_params["model_size"],choices=self.nllb_inf.available_models)
328
- dd_source_lang = gr.Dropdown(label="Source Language", value=nllb_params["source_lang"],choices=self.nllb_inf.available_source_langs)
329
- dd_target_lang = gr.Dropdown(label="Target Language", value=nllb_params["target_lang"],choices=self.nllb_inf.available_target_langs)
330
- with gr.Row():
331
- nb_max_length = gr.Number(label="Max Length Per Line", value=nllb_params["max_length"],precision=0)
332
- with gr.Row():
333
- cb_timestamp = gr.Checkbox(value=translation_params["add_timestamp"], label="Add a timestamp to the end of the filename",interactive=True)
334
- with gr.Row():
335
- btn_run = gr.Button("TRANSLATE SUBTITLE FILE", variant="primary")
336
- with gr.Row():
337
- tb_indicator = gr.Textbox(label="Output", scale=5)
338
- files_subtitles = gr.Files(label="Downloadable output file", scale=3)
339
- btn_openfolder = gr.Button('📂', scale=1)
340
-
341
- btn_run.click(fn=self.nllb_inf.translate_file,
342
- inputs=[file_subs, dd_model_size, dd_source_lang, dd_target_lang,
343
- nb_max_length, cb_timestamp],
344
- outputs=[tb_indicator, files_subtitles])
345
-
346
- btn_openfolder.click(fn=lambda: self.open_folder(os.path.join(self.args.output_dir, "translations")),
347
- inputs=None,
348
- outputs=None)
349
-
350
  # Launch the app with optional gradio settings
351
  args = self.args
352
 
 
102
  cb_timestamp = gr.Checkbox(value=whisper_params["add_timestamp"], label="Add timestamp to output file",interactive=True, visible=True)
103
  #dd_file_format = gr.Dropdown(choices=["SRT", "WebVTT", "txt"], value="SRT", label="File Format")
104
  dd_file_format = gr.Dropdown(choices=["TXT","SRT"], value="TXT", label="Output format", visible=False, info="Output preview format")
105
+
106
+ dd_model_size = gr.Dropdown(label="Model", value=nllb_params["model_size"],choices=self.nllb_inf.available_models)
107
+ dd_source_lang = gr.Dropdown(label="Source Language", value=nllb_params["source_lang"],choices=self.nllb_inf.available_source_langs)
108
+ dd_target_lang = gr.Dropdown(label="Target Language", value=nllb_params["target_lang"],choices=self.nllb_inf.available_target_langs)
109
+ nb_max_length = gr.Number(label="Max Length Per Line", value=nllb_params["max_length"],precision=0)
110
+
111
  # with gr.Accordion("Speaker diarization", open=False, visible=True):
112
  # cb_diarize = gr.Checkbox(value=diarization_params["is_diarize"], label="Use diarization",interactive=True)
113
  # tb_hf_token = gr.Text(label="Token", value=diarization_params["hf_token"],info="Required to use diarization")
 
324
  with gr.Column():
325
  gr.Markdown(device_info, label="Hardware info & installed packages")
326
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
327
  # Launch the app with optional gradio settings
328
  args = self.args
329