Update app.py

app.py

@@ -12,7 +12,7 @@ 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", device=0)
+translation_pipe = pipeline("translation", model="facebook/wmt19-en-ru", torch_dtype=torch.float16, device=0)
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()
 vae = get_vae().to(device)