Update app.py

app.py

@@ -3,10 +3,11 @@ import os
 
 os.system("pip install transformers sentencepiece torch")
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+from transformers import M2M100ForConditionalGeneration
+from tokenization_small100 import SMALL100Tokenizer
 
-tokenizer = AutoTokenizer.from_pretrained("alirezamsh/small100")
-model = AutoModelForSeq2SeqLM.from_pretrained("alirezamsh/small100")
+model = M2M100ForConditionalGeneration.from_pretrained("alirezamsh/small100")
+tokenizer = SMALL100Tokenizer.from_pretrained("alirezamsh/small100")
 
 def fn(text, lang):
     tokenizer.tgt_lang = lang