Hugging Face's logo

Spaces:
DoanMinhTri
/
MACHINE_TRANSLATION_CHARMED_ONE
Sleeping

App Files Community
DoanMinhTri commited on
Commit
f6640a9
•
1 Parent(s): 77e422b

init!

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -1,7 +1,7 @@
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
- md_name1 = "vinai/vinai-translate-vi2en-v2"
5
  tokenizer_vi2en = AutoTokenizer.from_pretrained(md_name1, src_lang="vi_VN")
6
  model_vi2en = AutoModelForSeq2SeqLM.from_pretrained(md_name1)
7
 
@@ -23,7 +23,7 @@ def translate_vi2en(vi_text: str) -> str:
23
  en_text = " ".join(en_text)
24
  return en_text
25
 
26
- md_name2 = "vinai/vinai-translate-en2vi-v2"
27
  tokenizer_en2vi = AutoTokenizer.from_pretrained(md_name2, src_lang="en_XX")
28
  model_en2vi = AutoModelForSeq2SeqLM.from_pretrained(md_name2)
29
 
 
1
  import gradio as gr
2
  from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
3
 
4
+ md_name1 = "DoanMinhTri/trilism_vi2en"
5
  tokenizer_vi2en = AutoTokenizer.from_pretrained(md_name1, src_lang="vi_VN")
6
  model_vi2en = AutoModelForSeq2SeqLM.from_pretrained(md_name1)
7
 
 
23
  en_text = " ".join(en_text)
24
  return en_text
25
 
26
+ md_name2 = "DoanMinhTri/trilism_en2vi"
27
  tokenizer_en2vi = AutoTokenizer.from_pretrained(md_name2, src_lang="en_XX")
28
  model_en2vi = AutoModelForSeq2SeqLM.from_pretrained(md_name2)
29