Update app.py
Browse files
app.py
CHANGED
|
@@ -24,7 +24,7 @@ from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
|
|
| 24 |
translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
| 25 |
translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
| 26 |
|
| 27 |
-
languages_list = [gu_IN, hi_IN ,bn_IN, ml_IN, mr_IN, ta_IN, te_IN]
|
| 28 |
|
| 29 |
def english_to_indian(sentence):
|
| 30 |
translation_tokenizer.src_lang = "en_xx"
|
|
|
|
| 24 |
translation_model = MBartForConditionalGeneration.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
| 25 |
translation_tokenizer = MBart50TokenizerFast.from_pretrained("facebook/mbart-large-50-many-to-many-mmt")
|
| 26 |
|
| 27 |
+
languages_list = ["gu_IN", "hi_IN" ,"bn_IN", "ml_IN", "mr_IN", "ta_IN", "te_IN"]
|
| 28 |
|
| 29 |
def english_to_indian(sentence):
|
| 30 |
translation_tokenizer.src_lang = "en_xx"
|