NLPV commited on
Commit
b589aed
·
verified ·
1 Parent(s): 4fe3864

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -50,7 +50,7 @@ def transcribe_and_translate(audio_file, selected_language, model_type="base"):
50
  end_time = segment['end']
51
  text = segment['text']
52
  text_file.write(f"[{start_time:.2f} - {end_time:.2f}] {text}\n")
53
- if selected_language in ['bn', 'mag', 'bho']:
54
  text_en = GoogleTranslator(source='auto', target='en').translate(text)
55
  translated_text.append(f"[{start_time:.2f} - {end_time:.2f}] {text_en}")
56
  text_file.write(f"[{start_time:.2f} - {end_time:.2f}] {text_en}\n")
@@ -66,11 +66,11 @@ interface = gr.Interface(
66
  fn=transcribe_and_translate,
67
  inputs=[
68
  gr.Audio(type="filepath", label="Upload Audio"),
69
- gr.Dropdown(label="Select Language", choices=["bn", "mag", "bho", "en"], value="mai"),
70
  gr.Dropdown(label="Select Model Type", choices=["tiny", "base", "small", "medium", "large"], value="base")
71
  ],
72
  outputs="text",
73
- title="Maithili, Maghi, and Bhojpuri Transcription and Translation"
74
  )
75
 
76
  if __name__ == '__main__':
 
50
  end_time = segment['end']
51
  text = segment['text']
52
  text_file.write(f"[{start_time:.2f} - {end_time:.2f}] {text}\n")
53
+ if selected_language in ['nl']:
54
  text_en = GoogleTranslator(source='auto', target='en').translate(text)
55
  translated_text.append(f"[{start_time:.2f} - {end_time:.2f}] {text_en}")
56
  text_file.write(f"[{start_time:.2f} - {end_time:.2f}] {text_en}\n")
 
66
  fn=transcribe_and_translate,
67
  inputs=[
68
  gr.Audio(type="filepath", label="Upload Audio"),
69
+ gr.Dropdown(label="Select Language", choices=["nl","en"], value="mai"),
70
  gr.Dropdown(label="Select Model Type", choices=["tiny", "base", "small", "medium", "large"], value="base")
71
  ],
72
  outputs="text",
73
+ title="Transcription and Translation"
74
  )
75
 
76
  if __name__ == '__main__':