sudoping01 commited on
Commit
e2be159
·
verified ·
1 Parent(s): defa96b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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="Bambara_Nko")
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...")