Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -14,7 +14,7 @@ tqdm.__init__ = partialmethod(tqdm.__init__, disable=True)
|
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16)
|
16 |
tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
|
17 |
-
translation_pipe = pipeline("translation", model=translation_model, device=0)
|
18 |
dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
|
19 |
tokenizer = get_tokenizer()
|
20 |
vae = get_vae().to(device)
|
|
|
14 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
15 |
translation_model = FSMTForConditionalGeneration.from_pretrained("facebook/wmt19-en-ru", torch_dtype=torch.float16)
|
16 |
tokenizer = FSMTTokenizer.from_pretrained("facebook/wmt19-en-ru")
|
17 |
+
translation_pipe = pipeline("translation", model=translation_model, tokenizer=tokenizer, device=0)
|
18 |
dalle = get_rudalle_model("Malevich", pretrained=True, fp16=True, device=device)
|
19 |
tokenizer = get_tokenizer()
|
20 |
vae = get_vae().to(device)
|