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_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16).to(device)
+translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16).half().to(device)
 translation_tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
 dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
 tokenizer = get_tokenizer()