ronaldahmed commited on
Commit
c2bcb6c
1 Parent(s): 7536bc5

Update ccl_win.py

Browse files
Files changed (1) hide show
  1. ccl_win.py +1 -1
ccl_win.py CHANGED
@@ -98,7 +98,7 @@ class Scorer:
98
  self.model_type = model_type
99
  self.batch_size = batch_size
100
  self._tokenizer = AutoTokenizer.from_pretrained("roberta-large")
101
- self._model = AutoModelForSequenceClassification.from_pretrained(model_type)
102
  self._model.to(device)
103
  self._model.eval()
104
 
 
98
  self.model_type = model_type
99
  self.batch_size = batch_size
100
  self._tokenizer = AutoTokenizer.from_pretrained("roberta-large")
101
+ self._model = AutoModelForSequenceClassification.from_pretrained("./" + model_type)
102
  self._model.to(device)
103
  self._model.eval()
104