Update app.py

app.py

@@ -2,7 +2,8 @@ import streamlit as st
 from io import BytesIO
 # import gradio as gr
 # Def_04 Docx file to translated_Docx file
-from transformers import MarianMTModel, MarianTokenizer
+#from transformers import MarianMTModel, MarianTokenizer
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 import nltk
 from nltk.tokenize import sent_tokenize
 from nltk.tokenize import LineTokenizer
@@ -55,8 +56,8 @@ def btTranslator(docxfile):
         st.spinner('Wait for it...')
         sleep(0.5)
         # mname = '/content/drive/MyDrive/Transformers Models/opus-mt-en-hi-Trans Model'
-        tokenizer = MarianTokenizer.from_pretrained(lang)
-        model = MarianMTModel.from_pretrained(lang)
+        tokenizer = AutoTokenizer.from_pretrained(lang)
+        model = AutoModelForSeq2SeqLM.from_pretrained(lang)
         model.to(device)
         lt = LineTokenizer()
         batch_size = 8