Spaces:

jeydipak
/

nlp_project

Runtime error

jeydipak commited on Mar 8, 2024

Commit

62bf081

verified ·

1 Parent(s): db07551

Update pages/task1.py

Files changed (1) hide show

pages/task1.py CHANGED Viewed

@@ -38,7 +38,7 @@ log_reg_bert = joblib.load('srcs/log_reg_bert.sav')
 texter = Text_ex(clean, vocab_to_int, SEQ_LEN)
 lstm = LSTMConcatAttentionB()
-lstm.load_state_dict(torch.load('srcs/lstm.pt'), map_location=torch.device('cpu'))
 vectorizer = pickle.load(open("srcs/vectorizer.pickle", "rb"))

 texter = Text_ex(clean, vocab_to_int, SEQ_LEN)
 lstm = LSTMConcatAttentionB()
+lstm.load_state_dict(torch.load('srcs/lstm.pt'))
 vectorizer = pickle.load(open("srcs/vectorizer.pickle", "rb"))