Update app.py
Browse files
app.py
CHANGED
@@ -14,8 +14,8 @@ from transformers import pipeline
|
|
14 |
import gradio as gr
|
15 |
|
16 |
# Define source and target language
|
17 |
-
src_lang = "
|
18 |
-
tgt_lang = "
|
19 |
|
20 |
# Load model with language codes
|
21 |
model = pipeline("translation", model="BeebekBhz/Final-ne-en", src_lang=src_lang, tgt_lang=tgt_lang)
|
@@ -24,4 +24,4 @@ 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)
|
|
|
14 |
import gradio as gr
|
15 |
|
16 |
# Define source and target language
|
17 |
+
src_lang = "ne" # Nepali
|
18 |
+
tgt_lang = "en" # English
|
19 |
|
20 |
# Load model with language codes
|
21 |
model = pipeline("translation", model="BeebekBhz/Final-ne-en", src_lang=src_lang, tgt_lang=tgt_lang)
|
|
|
24 |
return model(text)[0]['translation_text']
|
25 |
|
26 |
iface = gr.Interface(fn=translate, inputs="text", outputs="text")
|
27 |
+
iface.launch(share=True)
|