Spaces:

msarmi9
/

multi30k

Runtime error

msarmi9 commited on Feb 17, 2022

Commit

2afffb1

1 Parent(s): e40e595

[update]: huggingface hub model path

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ from models import Seq2Seq
 # Load model
-model_path = hf_hub_download("msarmi9/multi30k", "models/de-en/model.bin")
 model = Seq2Seq(vocab_size=8000, hidden_dim=512, bos_idx=1, eos_idx=2, pad_idx=3, temperature=2)
 model.load_state_dict(torch.load(model_path))
 model.eval()

 # Load model
+model_path = hf_hub_download("msarmi9/multi30k", "models/de-en/version_1/model.bin")
 model = Seq2Seq(vocab_size=8000, hidden_dim=512, bos_idx=1, eos_idx=2, pad_idx=3, temperature=2)
 model.load_state_dict(torch.load(model_path))
 model.eval()