Update app.py

app.py

@@ -1,10 +1,11 @@
 import gradio as gr
 
-from transformers import M2M100ForConditionalGeneration
-from tokenization_small100 import SMALL100Tokenizer
+os.system("pip install transformers")
 
-model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
-tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
+from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+tokenizer = AutoTokenizer.from_pretrained("alirezamsh/small100")
+model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
 
 def fn(text):
     tokenizer.tgt_lang = "en"