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
+from transformers import pipeline, FSMTForConditionalGeneration
 from rudalle.pipelines import generate_images
 from rudalle import get_rudalle_model, get_tokenizer, get_vae
 
@@ -12,7 +12,8 @@ from rudalle import get_rudalle_model, get_tokenizer, get_vae
 tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
 
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
-translation_pipe = pipeline("translation", model="facebook/wmt19-en-ru", torch_dtype=torch.float16, device=0)
+translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16)
+translation_pipe = pipeline("translation", model=translation_model, device=0)
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae().to(device)