zoya23 commited on
Commit
0c89568
·
verified ·
1 Parent(s): e9fd467

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -3,10 +3,10 @@ from transformers import pipeline
3
 
4
  # Language pair to model mapping
5
  model_mapping = {
6
- ("English", "Hindi"):"google-bert/bert-base-multilingual-cased", # "Helsinki-NLP/opus-mt-en-hi",
7
- ("Hindi", "English"): "google-bert/bert-base-multilingual-cased", # "TestZee/Finetuned-hindi-to-english-V5",
8
- ("English", "Telugu"):"google-bert/bert-base-multilingual-cased", # "harshitha2406/English_to_Telugu",
9
- ("Telugu", "English"):"google-bert/bert-base-multilingual-cased", # "ai4bharat/indic-translator-te-en",
10
  }
11
 
12
  # Streamlit page config
 
3
 
4
  # Language pair to model mapping
5
  model_mapping = {
6
+ ("English", "Hindi"):"facebook/mbart-large-50-many-to-many-mmt", # "Helsinki-NLP/opus-mt-en-hi",
7
+ ("Hindi", "English"): "facebook/mbart-large-50-many-to-many-mmt", # "TestZee/Finetuned-hindi-to-english-V5",
8
+ ("English", "Telugu"):"facebook/mbart-large-50-many-to-many-mmt", # "harshitha2406/English_to_Telugu",
9
+ ("Telugu", "English"):"facebook/mbart-large-50-many-to-many-mmt", # "ai4bharat/indic-translator-te-en",
10
  }
11
 
12
  # Streamlit page config