lampongyuen commited on
Commit
d994ddf
·
1 Parent(s): 86b10da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -6,7 +6,10 @@ import torch
6
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
7
  tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
8
  device = 0 if torch.cuda.is_available() else -1
9
- LANGS = ["ace_Arab", "eng_Latn", "fra_Latn", "spa_Latn"]
 
 
 
10
 
11
  def translate(text, src_lang, tgt_lang):
12
  """
 
6
  model = AutoModelForSeq2SeqLM.from_pretrained("facebook/nllb-200-distilled-600M")
7
  tokenizer = AutoTokenizer.from_pretrained("facebook/nllb-200-distilled-600M")
8
  device = 0 if torch.cuda.is_available() else -1
9
+ LANGS = ["ace_Arab", "eng_Latn", "fra_Latn", "spa_Latn", "yue_Hant","Zho_Hans","Zho_Hant"]
10
+
11
+ # Yue Chinese - yue_Hant, Chinese (Simplified)-Zho_Hans, Chinese(Traditional)-zho_Hant
12
+ # https://github.com/facebookresearch/flores/tree/main/flores200#languages-in-flores-200
13
 
14
  def translate(text, src_lang, tgt_lang):
15
  """