StevanLS commited on
Commit
875d135
·
1 Parent(s): 96df7fd

Language detection feature

Browse files
Files changed (2) hide show
  1. app.py +6 -4
  2. requirements.txt +2 -1
app.py CHANGED
@@ -1,16 +1,18 @@
1
  import translators as ts
2
  import gradio as gr
 
3
 
4
- def translate_transformers(input_language, output_language, input_text,):
5
  languages_dict = {'Spanish': 'es', 'Bengali': 'bn', 'English': 'en', 'French': 'fr'}
 
 
 
6
  return ts.translate_text(input_text, translator="google",
7
- from_language=languages_dict[input_language],
8
  to_language=languages_dict[output_language])
9
 
10
  interface = gr.Interface(fn=translate_transformers,
11
  inputs=[
12
- gr.Dropdown(['Spanish', 'Bengali', 'English', 'French'],
13
- label='Input language'),
14
  gr.Dropdown(['Spanish', 'Bengali', 'English', 'French'],
15
  label = 'Translated language'),
16
  gr.Textbox(lines=2, placeholder='Text to translate',
 
1
  import translators as ts
2
  import gradio as gr
3
+ from langdetect import detect
4
 
5
+ def translate_transformers(output_language, input_text,):
6
  languages_dict = {'Spanish': 'es', 'Bengali': 'bn', 'English': 'en', 'French': 'fr'}
7
+
8
+ input_language = detect(input_text)
9
+
10
  return ts.translate_text(input_text, translator="google",
11
+ from_language=input_language,
12
  to_language=languages_dict[output_language])
13
 
14
  interface = gr.Interface(fn=translate_transformers,
15
  inputs=[
 
 
16
  gr.Dropdown(['Spanish', 'Bengali', 'English', 'French'],
17
  label = 'Translated language'),
18
  gr.Textbox(lines=2, placeholder='Text to translate',
requirements.txt CHANGED
@@ -1 +1,2 @@
1
- translators
 
 
1
+ translators
2
+ langdetect