Update app.py

app.py

@@ -1,15 +1,16 @@
 import streamlit as st
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
-from datasets import load_dataset
+from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
 
-# Login using e.g. `huggingface-cli login` to access this dataset
-ds = load_dataset("pavanteja15/english-to-telugu-570K")  
-tokenizer = AutoTokenizer.from_pretrained(model_name)
-model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
+# Define model name
+model_checkpoint = "aryaumesh/english-to-telugu"
+
+# Load tokenizer and model
+tokenizer = MBart50TokenizerFast.from_pretrained(model_checkpoint)
+model = MBartForConditionalGeneration.from_pretrained(model_checkpoint)
 
 # Streamlit UI
-st.title("Colloquial Language Translator")
-st.write("Enter English text to translate into the colloquial language of your choice.")
+st.title("English to Telugu Translator")
+st.write("Enter English text to translate into Telugu.")
 
 # User input
 input_text = st.text_input("Enter English text:")