Update app.py

app.py

@@ -4,7 +4,7 @@ from tqdm import tqdm
 from functools import partialmethod
 import gradio as gr
 from gradio.mix import Series
-from transformers import pipeline, FSMTForConditionalGeneration
+from transformers import pipeline, FSMTForConditionalGeneration, FSMTTokenizer
 from rudalle.pipelines import generate_images
 from rudalle import get_rudalle_model, get_tokenizer, get_vae
 
@@ -13,6 +13,7 @@ tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16)
+tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
 translation_pipe = pipeline("translation", model=translation_model, device=0)
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()