Spaces:

nkigumnov
/

banks-ethics-sentiment

Build error

nkigumnov commited on Apr 17, 2023

Commit

9cbe19c

1 Parent(s): 6078972

Update model.py

Files changed (1) hide show

model.py CHANGED Viewed

@@ -122,7 +122,7 @@ class CustomOrdinalLogisticModel(nn.Module):
 tokenizer = AutoTokenizer.from_pretrained('blanchefort/rubert-base-cased-sentiment-rusentiment')
-model = CustomOrdinalLogisticModel(Union_model('blanchefort/rubert-base-cased-sentiment-rusentiment'), 3).to('cpu')
 model.load_state_dict(torch.load('best_model.pth', map_location='cpu'), strict=False)

 tokenizer = AutoTokenizer.from_pretrained('blanchefort/rubert-base-cased-sentiment-rusentiment')
+model = CustomOrdinalLogisticModel(UnionModel('blanchefort/rubert-base-cased-sentiment-rusentiment'), 3).to('cpu')
 model.load_state_dict(torch.load('best_model.pth', map_location='cpu'), strict=False)