BeebekBhz commited on
Commit
83daf67
·
verified ·
1 Parent(s): 7da6382

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -2
app.py CHANGED
@@ -13,10 +13,15 @@ except ImportError:
13
  from transformers import pipeline
14
  import gradio as gr
15
 
16
- model = pipeline("translation", model="BeebekBhz/Final-en-ne")
 
 
 
 
 
17
 
18
  def translate(text):
19
  return model(text)[0]['translation_text']
20
 
21
  iface = gr.Interface(fn=translate, inputs="text", outputs="text")
22
- iface.launch()
 
13
  from transformers import pipeline
14
  import gradio as gr
15
 
16
+ # Define source and target language
17
+ src_lang = "eng_Latn" # English
18
+ tgt_lang = "npi_Deva" # Nepali (Devanagari script)
19
+
20
+ # Load model with language codes
21
+ model = pipeline("translation", model="BeebekBhz/Final-en-ne", src_lang=src_lang, tgt_lang=tgt_lang)
22
 
23
  def translate(text):
24
  return model(text)[0]['translation_text']
25
 
26
  iface = gr.Interface(fn=translate, inputs="text", outputs="text")
27
+ iface.launch(share=True)