Update app.py

app.py

@@ -1,17 +1,18 @@
 import streamlit as st
 from transformers import MarianMTModel, MarianTokenizer
 
-# Load the pre-trained MarianMT model and tokenizer
+# Load the MarianMT model and tokenizer
 model_name = "Helsinki-NLP/opus-mt-en-ROMANCE"
 model = MarianMTModel.from_pretrained(model_name)
 tokenizer = MarianTokenizer.from_pretrained(model_name)
 
 def translate_text(text, target_lang):
-    # Translate English to the target language
+    # Prepare the input and translate
     inputs = tokenizer.encode(text, return_tensors="pt")
     translated = model.generate(inputs, decoder_start_token_id=tokenizer.lang_code_to_id[target_lang])
     translated_text = tokenizer.decode(translated[0], skip_special_tokens=True)
     return translated_text
+
 def main():
     st.title("English to Any Language Translator")