Update app.py

app.py

@@ -1,15 +1,12 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the translation pipeline from Hugging Face
-translator = pipeline("translation_en_to_fr")  # Default to French translation
-
 # Available languages and their corresponding models
 available_languages = {
-    "French": "translation_en_to_fr",
-    "German": "translation_en_to_de",
-    "Spanish": "translation_en_to_es",
-    "Chinese": "Helsinki-NLP/opus-mt-en-zh",  # For Chinese, we use a specific model from Hugging Face
+    "French": "Helsinki-NLP/opus-mt-en-fr",
+    "German": "Helsinki-NLP/opus-mt-en-de",
+    "Spanish": "Helsinki-NLP/opus-mt-en-es",
+    "Chinese": "Helsinki-NLP/opus-mt-en-zh",
     "Japanese": "Helsinki-NLP/opus-mt-en-jap",
     "Russian": "Helsinki-NLP/opus-mt-en-ru",
     "Arabic": "Helsinki-NLP/opus-mt-en-ar",
@@ -24,7 +21,7 @@ text_to_translate = st.text_area("Enter text in English:", "")
 # Language selection
 target_language = st.selectbox("Select the target language:", list(available_languages.keys()))
 
-# Change the translation model based on the selected language
+# Load the translation model based on the selected language
 translator = pipeline("translation", model=available_languages[target_language])
 
 # Translate button