Sid26Roy fischerman commited on
Commit
ebf0184
·
verified ·
1 Parent(s): 4785126

Update app.py (#1)

Browse files

- Update app.py (6fd26a9eb2e11f1132a1b6d94d88276dbd9fe84b)


Co-authored-by: cooper <fischerman@users.noreply.huggingface.co>

Files changed (1) hide show
  1. app.py +19 -3
app.py CHANGED
@@ -14,9 +14,25 @@ tokenizer = AutoTokenizer.from_pretrained("ai4bharat/indictrans2-en-indic-dist-2
14
  model = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/indictrans2-en-indic-dist-200M", trust_remote_code=True)
15
 
16
  LANG_OPTIONS = [
17
- "hin_Deva", "ben_Beng", "pan_Guru", "guj_Gujr",
18
- "tam_Taml", "tel_Telu", "mal_Mlym",
19
- "mar_Deva", "kan_Knda", "asm_Beng"
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
20
  ]
21
 
22
  def translate(text, target_lang):
 
14
  model = AutoModelForSeq2SeqLM.from_pretrained("ai4bharat/indictrans2-en-indic-dist-200M", trust_remote_code=True)
15
 
16
  LANG_OPTIONS = [
17
+ "hin_Deva", # Hindi
18
+ "ben_Beng", # Bengali
19
+ "pan_Guru", # Punjabi
20
+ "guj_Gujr", # Gujarati
21
+ "tam_Taml", # Tamil
22
+ "tel_Telu", # Telugu
23
+ "mal_Mlym", # Malayalam
24
+ "mar_Deva", # Marathi
25
+ "kan_Knda", # Kannada
26
+ "asm_Beng", # Assamese
27
+ "kas_Arab", # Kashmiri (Arabic)
28
+ "kas_Deva", # Kashmiri (Devanagari)
29
+ "san_Deva", # Sanskrit
30
+ "brx_Deva", # Bodo
31
+ "mai_Deva", # Maithili
32
+ "sat_Olck", # Santali
33
+ "eng_Latn", # English
34
+ "urd_Arab" # Urdu
35
+
36
  ]
37
 
38
  def translate(text, target_lang):