AreesaAshfaq commited on
Commit
6202397
·
verified ·
1 Parent(s): bcccee3

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +43 -0
app.py ADDED
@@ -0,0 +1,43 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import MarianMTModel, MarianTokenizer
3
+
4
+ # Load the pre-trained MarianMT model and tokenizer
5
+ model_name = "Helsinki-NLP/opus-mt-en-ROMANCE"
6
+ model = MarianMTModel.from_pretrained(model_name)
7
+ tokenizer = MarianTokenizer.from_pretrained(model_name)
8
+
9
+ def translate_text(text, target_lang):
10
+ # Translate English to the target language
11
+ inputs = tokenizer.encode(text, return_tensors="pt")
12
+ translated = model.generate(inputs, decoder_start_token_id=tokenizer.lang_code_to_id[target_lang])
13
+ translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
14
+ return translated_text
15
+ def main():
16
+ st.title("English to Any Language Translator")
17
+
18
+ # User input for English text
19
+ text_to_translate = st.text_area("Enter text in English:")
20
+
21
+ # Language options
22
+ languages = {
23
+ 'French': 'fr',
24
+ 'German': 'de',
25
+ 'Italian': 'it',
26
+ 'Portuguese': 'pt',
27
+ 'Spanish': 'es',
28
+ # Add more languages if needed
29
+ }
30
+
31
+ selected_language = st.selectbox("Select target language:", list(languages.keys()))
32
+
33
+ if st.button("Translate"):
34
+ if text_to_translate:
35
+ target_lang = languages[selected_language]
36
+ translated_text = translate_text(text_to_translate, target_lang)
37
+ st.write(f"**Translation in {selected_language}:**")
38
+ st.write(translated_text)
39
+ else:
40
+ st.warning("Please enter text to translate.")
41
+
42
+ if __name__ == "__main__":
43
+ main()