Spaces:

osheina
/

NLP_project

Sleeping

osheina commited on Apr 12, 2024

Commit

f3d621f

verified ·

1 Parent(s): 3fa3430

Update bert_strim.py

Files changed (1) hide show

bert_strim.py CHANGED Viewed

@@ -8,7 +8,7 @@ tokenizer = AutoTokenizer.from_pretrained("cointegrated/rubert-tiny2")
 model = BERTClassifier()
 device = 'cpu'
-model.load_state_dict(torch.load('/Users/olgaseina/Desktop/NLP_project/BERTmodel_weights2.pth',map_location=torch.device('cpu')))
 model.eval()
 @st.cache_data

 model = BERTClassifier()
 device = 'cpu'
+model.load_state_dict(torch.load('BERTmodel_weights2.pth',map_location=torch.device('cpu')))
 model.eval()
 @st.cache_data