Update app.py

app.py

@@ -1,49 +1,49 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
-# Sepecified different languages
+# Specified different languages
 LANGUAGES = {
-    "English" : "en",
-    "French" : 'fr',
-    "German" : "de",
-    "Spanish" : "es",
-    "Italian" : "it",
-    "Portuguese" : "pt",
-    "Russian" : "ru",
-    "Chinese" : "zh",
-    "Japense" : "ja",
-    "Arabic" : "ar",
+    "English": "en",
+    "French": "fr",
+    "German": "de",
+    "Spanish": "es",
+    "Italian": "it",
+    "Portuguese": "pt",
+    "Russian": "ru",
+    "Chinese": "zh",
+    "Japanese": "ja",
+    "Arabic": "ar",
 }
 
 # Helper function to load the model and tokenizer
 @st.cache_resource
-def load_translation_model(src_lang, tgt_lang):
-    model_name = f"Helsinki-NLP.opus-mt-{src_lang}-{tgt-lang}"
+def load_translation_model(src_lang, tgt_lang):  # Corrected parameter names
+    model_name = f"Helsinki-NLP/opus-mt-{src_lang}-{tgt_lang}"  # Corrected typo in model_name
     tokenizer = MarianTokenizer.from_pretrained(model_name)
     model = MarianMTModel.from_pretrained(model_name)
-    return tokenizer,model
+    return tokenizer, model
 
 # Translation function
 def translate_text(tokenizer, model, text):
     inputs = tokenizer(text, return_tensors='pt', padding=True)
     translated = model.generate(**inputs)
-    return tokenizer.decode(translated[0],skip_special_tokens = True)
+    return tokenizer.decode(translated[0], skip_special_tokens=True)
 
-#Streamlit App
+# Streamlit App
 st.title("Language Translation APP 🌍")
-st.write("Translate text between multiple language using open-source models.")
+st.write("Translate text between multiple languages using open-source models.")
 
-#Language selection
+# Language selection
 col1, col2 = st.columns(2)
 with col1:
     source_language = st.selectbox("Select Source Language 🌐", list(LANGUAGES.keys()))
 with col2:
-    target_language = st.selectbox("Select Target Language 🌍",list(LANGUAGES.keys()))
+    target_language = st.selectbox("Select Target Language 🌍", list(LANGUAGES.keys()))
 
-#Input Text
-text_to_translate  = st.text_area("Enter text to translate", height=150)
+# Input Text
+text_to_translate = st.text_area("Enter text to translate", height=150)
 
-#Translate button
+# Translate button
 if st.button("Translate"):
     if source_language == target_language:
         st.warning("Source and target language must be different")
@@ -54,16 +54,15 @@ if st.button("Translate"):
         tgt_lang = LANGUAGES[target_language]
 
         try:
-            #Load model and tokenizer
+            # Load model and tokenizer
             tokenizer, model = load_translation_model(src_lang, tgt_lang)
 
-            #Perform translation
+            # Perform translation
             translated_text = translate_text(tokenizer, model, text_to_translate)
 
             # Display result
-            st.subheader("Translate Text 🔄: ")
-            st.write(translate_text)
+            st.subheader("Translated Text 🔄:")
+            st.write(translated_text)
 
         except Exception as e:
-            st.error(f"Error : {str(e)}")
-        
+            st.error(f"Error: {str(e)}")