Sukumar2005 commited on
Commit
0ef4e08
Β·
verified Β·
1 Parent(s): 675da53

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -4,17 +4,17 @@ import gradio as gr
4
  # Load translation pipelines once
5
  t5_translator = pipeline("translation_en_to_de", model="t5-small")
6
 
7
- # βœ… Correct model name for English β†’ Tamil
8
  ta_translator = pipeline("translation_en_to_ta", model="Helsinki-NLP/opus-mt-en-tam")
9
 
10
  def translate_text(text):
11
- # Translate using T5-small
12
  result_t5 = t5_translator(text, max_length=40)[0]['translation_text']
13
- # Translate using Helsinki-NLP English β†’ Tamil
14
  result_ta = ta_translator(text, max_length=40)[0]['translation_text']
15
  return result_t5, result_ta
16
 
17
- # Gradio interface
18
  demo = gr.Interface(
19
  fn=translate_text,
20
  inputs=gr.Textbox(lines=2, placeholder="Enter English text here..."),
 
4
  # Load translation pipelines once
5
  t5_translator = pipeline("translation_en_to_de", model="t5-small")
6
 
7
+ # βœ… Correct and public model for English β†’ Tamil
8
  ta_translator = pipeline("translation_en_to_ta", model="Helsinki-NLP/opus-mt-en-tam")
9
 
10
  def translate_text(text):
11
+ # Translate using T5-small (English β†’ German)
12
  result_t5 = t5_translator(text, max_length=40)[0]['translation_text']
13
+ # Translate using Helsinki-NLP (English β†’ Tamil)
14
  result_ta = ta_translator(text, max_length=40)[0]['translation_text']
15
  return result_t5, result_ta
16
 
17
+ # Create Gradio interface
18
  demo = gr.Interface(
19
  fn=translate_text,
20
  inputs=gr.Textbox(lines=2, placeholder="Enter English text here..."),