sarahai commited on
Commit
752c689
1 Parent(s): 7d91da2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +8 -8
app.py CHANGED
@@ -1,25 +1,25 @@
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
3
 
4
- # Load the pre-trained model and tokenizer
5
  model_name = "sarahai/nllb-ru-uz"
6
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
  tokenizer = NllbTokenizer.from_pretrained(model_name)
8
 
9
  def translate(text, src_lang="rus_Cyrl", tgt_lang="uzn_Latn"):
10
- """Translates text from source to target language."""
11
  inputs = tokenizer(text, return_tensors="pt")
12
  translated = model.generate(**inputs)
13
  return tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
14
 
15
- # Define the Gradio interface
16
  interface = gr.Interface(
17
  fn=translate,
18
- inputs=[gr.Textbox(label="Text to Translate"), gr.Dropdown(choices=["ru", "uz"], label="Source Language"), gr.Dropdown(choices=["uz"], label="Target Language")],
19
  outputs="textbox",
20
- title="Russian to Uzbek Translator",
21
- description="Translate text from Russian to Uzbek using the `sarahai/nllb-ru-uz` model.",
22
  )
23
 
24
- # Launch the Gradio app
25
- interface.launch(share=True, debug=True) # Set share=True to create a Hugging Face Space
 
1
  import gradio as gr
2
  from transformers import AutoModelForSeq2SeqLM, NllbTokenizer
3
 
4
+
5
  model_name = "sarahai/nllb-ru-uz"
6
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
  tokenizer = NllbTokenizer.from_pretrained(model_name)
8
 
9
  def translate(text, src_lang="rus_Cyrl", tgt_lang="uzn_Latn"):
10
+
11
  inputs = tokenizer(text, return_tensors="pt")
12
  translated = model.generate(**inputs)
13
  return tokenizer.batch_decode(translated, skip_special_tokens=True)[0]
14
 
15
+
16
  interface = gr.Interface(
17
  fn=translate,
18
+ inputs=[gr.Textbox(label="Text to Translate")],
19
  outputs="textbox",
20
+ title="Русско-Узбекский Переводчик",
21
+ description="Перевод русских текстов на узбекский с помощью `sarahai/nllb-ru-uz` model.",
22
  )
23
 
24
+
25
+ interface.launch(share=True, debug=True)