Spaces:

ruslanruslanruslan
/

nlp_project

Sleeping

ruslanruslanruslan commited on Jul 21, 2023

Commit

0f494d0

•

1 Parent(s): 3ffa1c6

errors fixed

Files changed (1) hide show

pages/Oxymiron.py CHANGED Viewed

@@ -9,7 +9,7 @@ def load_model():
     return oxymiron, tokenizer
 oxymiron, tokenizer = load_model()
-oxymiron.load_state_dict(torch.load('pushkin_weights.pt', map_location=torch.device('cpu')))
 oxymiron.to('cpu')
 oxymiron.eval()

     return oxymiron, tokenizer
 oxymiron, tokenizer = load_model()
+oxymiron.load_state_dict(torch.load('oxymiron_weights.pt', map_location=torch.device('cpu')))
 oxymiron.to('cpu')
 oxymiron.eval()