anton-l HF staff commited on
Commit
e6ae6ee
·
1 Parent(s): 27cb35e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)