Hugging Face's logo

Spaces:
Ruslan-DS
/
nlp-project
Running

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

Update models/BertTunning.py

Browse files
Files changed (1) hide show
  1. models/BertTunning.py +1 -1
models/BertTunning.py CHANGED
@@ -33,7 +33,7 @@ class BertTunnig(nn.Module):
33
 
34
 
35
  model_tunning = BertTunnig(bert_model=model)
36
- model_tunning.load_state_dict(torch.load('models/weights/BertTunnigWeights.pt', map_location=torch.device('cpu')))))
37
 
38
 
39
  def predict_2(text):
 
33
 
34
 
35
  model_tunning = BertTunnig(bert_model=model)
36
+ model_tunning.load_state_dict(torch.load('models/weights/BertTunnigWeights.pt', map_location=torch.device('cpu')))
37
 
38
 
39
  def predict_2(text):