Update app.py

app.py

@@ -1,12 +1,14 @@
 import gradio as gr
-from transformers import pipeline
+from transformers import pipeline, AutoTokenizer, AutoModelForSeq2SeqLM
 
-pipe = pipeline("translation", model="csebuetnlp/banglat5_nmt_bn_en")
+tokenizer = AutoTokenizer.from_pretrained("csebuetnlp/banglat5_nmt_bn_en", use_fast=False)
+model = AutoModelForSeq2SeqLM.from_pretrained("csebuetnlp/banglat5_nmt_bn_en")
+
+pipe = pipeline("translation", model=model, tokenizer=tokenizer)
 
 def translate(text):
     translation = pipe(text)[0]['translation_text']
     return translation
-
 interface = gr.Interface(
     fn=translate,
     inputs=gr.inputs.Textbox(label="Input Text in Bangla"),
@@ -15,4 +17,4 @@ interface = gr.Interface(
     description="Translate Bangla text into English using a Hugging Face model."
 )
 if __name__ == "__main__":
-    interface.launch()
+    interface.launch()