zoya23 commited on
Commit
8834990
·
verified ·
1 Parent(s): e70601b

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -5
app.py CHANGED
@@ -3,17 +3,15 @@ from transformers import pipeline
3
 
4
  # Language pair to model mapping
5
  model_mapping = {
6
- ("English", "Hindi"):"Helsinki-NLP/opus-mt-en-hi", # "Helsinki-NLP/opus-mt-en-hi",
7
- ("Hindi", "English"): "TestZee/Finetuned-hindi-to-english-V5", # "TestZee/Finetuned-hindi-to-english-V5",
8
- ("English", "Telugu"):"hima06varshini/TeluguTranslatica", # "harshitha2406/English_to_Telugu",not working
9
- ("Telugu", "English"):"hima06varshini/TeluguTranslatica", # "ai4bharat/indic-translator-te-en",not working
10
  }
11
 
12
  # Streamlit page config
13
  st.set_page_config(page_title="Translator", layout="wide")
14
 
15
  # Only allowed languages
16
- languages = ["English", "Hindi", "Telugu"]
17
 
18
  # UI
19
  st.markdown("""
 
3
 
4
  # Language pair to model mapping
5
  model_mapping = {
6
+ ("English", "Hindi"):"zoya23/en-hi-translation # "Helsinki-NLP/opus-mt-en-hi",
7
+ ("Hindi", "English"): "TestZee/Finetuned-hindi-to-english-V5", # "TestZee/Finetuned-hindi-to-english-V5"
 
 
8
  }
9
 
10
  # Streamlit page config
11
  st.set_page_config(page_title="Translator", layout="wide")
12
 
13
  # Only allowed languages
14
+ languages = ["English", "Hindi"]
15
 
16
  # UI
17
  st.markdown("""