test

app.py

@@ -2,12 +2,12 @@
 baseline_interactive.py
 """
 import gradio as gr
-from transformers import MBartForConditionalGeneration, MBart50TokenizerFast
+from transformers import MBartForConditionalGeneration, MBartTokenizer
 from transformers import pipeline
 
 model_name = "momo/rsp-sum"
 model = MBartForConditionalGeneration.from_pretrained(model_name)
-tokenizer = MBart50TokenizerFast.from_pretrained(model_name, src_lang="ko_KR", tgt_lang="ko_KR")
+tokenizer = MBartTokenizer.from_pretrained(model_name, src_lang="ko_KR", tgt_lang="ko_KR")
 
 # prefix = "translate English to German: "