Abbas133 commited on
Commit
8575f8b
·
verified ·
1 Parent(s): ac092ce

Create app.py

Browse files
Files changed (1) hide show
  1. app.py +47 -0
app.py ADDED
@@ -0,0 +1,47 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ import streamlit as st
2
+ from transformers import M2M100ForConditionalGeneration, M2M100Tokenizer
3
+
4
+ # Load the model and tokenizer
5
+ model_name = "facebook/m2m100_418M"
6
+ tokenizer = M2M100Tokenizer.from_pretrained(model_name)
7
+ model = M2M100ForConditionalGeneration.from_pretrained(model_name)
8
+
9
+ # Streamlit UI
10
+ st.title("English to Multiple Language Translator")
11
+ st.write("Translate English text into different languages using AI.")
12
+
13
+ # Input text
14
+ input_text = st.text_area("Enter English text:", value="")
15
+
16
+ # Language selection
17
+ language_options = {
18
+ "French": "fr",
19
+ "Spanish": "es",
20
+ "German": "de",
21
+ "Chinese": "zh",
22
+ "Arabic": "ar",
23
+ "Hindi": "hi",
24
+ "Japanese": "ja",
25
+ "Russian": "ru",
26
+ "Portuguese": "pt",
27
+ "Italian": "it"
28
+ }
29
+ selected_language = st.selectbox("Select target language:", list(language_options.keys()))
30
+
31
+ if st.button("Translate"):
32
+ if input_text:
33
+ # Set target language
34
+ target_language = language_options[selected_language]
35
+ tokenizer.src_lang = "en"
36
+ encoded_input = tokenizer(input_text, return_tensors="pt")
37
+
38
+ # Generate translation
39
+ generated_tokens = model.generate(**encoded_input, forced_bos_token_id=tokenizer.get_lang_id(target_language))
40
+ translated_text = tokenizer.batch_decode(generated_tokens, skip_special_tokens=True)[0]
41
+
42
+ # Display translated text
43
+ st.write(f"**Translated text ({selected_language}):**")
44
+ st.write(translated_text)
45
+ else:
46
+ st.write("Please enter text to translate.")
47
+