Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
|
@@ -3,7 +3,7 @@ from transformers import pipeline
|
|
| 3 |
from typing import Union, List
|
| 4 |
|
| 5 |
class BambaraTranslator:
|
| 6 |
-
def __init__(self, model_name: str ="sudoping01/nllb-200-600M-bambara-nko" ): #"sudoping01/nllb-bambara-v2"):
|
| 7 |
self.translator = pipeline(
|
| 8 |
"translation",
|
| 9 |
model=model_name,
|
|
@@ -64,7 +64,7 @@ with gr.Blocks(title="Bambara Translator") as demo:
|
|
| 64 |
)
|
| 65 |
|
| 66 |
with gr.Row():
|
| 67 |
-
src_lang = gr.Dropdown(choices=["Bambara_Latin", "French", "English", "Bambara_Nko"], label="Source Language", value="
|
| 68 |
tgt_lang = gr.Dropdown(choices=["Bambara_Latin", "French", "English", "Bambara_Nko"], label="Target Language", value="French")
|
| 69 |
with gr.Column():
|
| 70 |
text_input = gr.Textbox(label="Input Text", lines=5, placeholder="Type your text here...")
|
|
|
|
| 3 |
from typing import Union, List
|
| 4 |
|
| 5 |
class BambaraTranslator:
|
| 6 |
+
def __init__(self, model_name: str ="sudoping01/nllb-bambara-v2"): #"sudoping01/nllb-200-600M-bambara-nko" ): #"sudoping01/nllb-bambara-v2"):
|
| 7 |
self.translator = pipeline(
|
| 8 |
"translation",
|
| 9 |
model=model_name,
|
|
|
|
| 64 |
)
|
| 65 |
|
| 66 |
with gr.Row():
|
| 67 |
+
src_lang = gr.Dropdown(choices=["Bambara_Latin", "French", "English", "Bambara_Nko"], label="Source Language", value="Bambara_Latin")
|
| 68 |
tgt_lang = gr.Dropdown(choices=["Bambara_Latin", "French", "English", "Bambara_Nko"], label="Target Language", value="French")
|
| 69 |
with gr.Column():
|
| 70 |
text_input = gr.Textbox(label="Input Text", lines=5, placeholder="Type your text here...")
|