Hugging Face's logo

Spaces:
Ruslan-DS
/
nlp-project
Running

App Files Community
Ruslan-DS commited on
Commit
ec33c28
•
1 Parent(s): d832547

Update models/LSTM.py

Browse files
Files changed (1) hide show
  1. models/LSTM.py +1 -1
models/LSTM.py CHANGED
@@ -73,7 +73,7 @@ class LSTMnn(nn.Module):
73
 
74
 
75
  model = LSTMnn()
76
- model.load_state_dict(torch.load('models/weights/LSTMBestWeights.pt', map_location=torch.device('cpu')))))
77
 
78
 
79
  def predict_3(text):
 
73
 
74
 
75
  model = LSTMnn()
76
+ model.load_state_dict(torch.load('models/weights/LSTMBestWeights.pt', map_location=torch.device('cpu')))
77
 
78
 
79
  def predict_3(text):