ronaldahmed commited on
Commit
8d6d100
1 Parent(s): 6314abb

Update ccl_win.py

Browse files
Files changed (1) hide show
  1. ccl_win.py +1 -1
ccl_win.py CHANGED
@@ -96,7 +96,7 @@ class Scorer:
96
  self.model_type = model_type
97
  self.batch_size = batch_size
98
  self._tokenizer = AutoTokenizer.from_pretrained("roberta-large")
99
- self._model = AutoModelForSequenceClassification.from_pretrained(os.path.join(os.getcwd(),model_type))
100
  self._model.to(device)
101
  self._model.eval()
102
 
 
96
  self.model_type = model_type
97
  self.batch_size = batch_size
98
  self._tokenizer = AutoTokenizer.from_pretrained("roberta-large")
99
+ self._model = AutoModelForSequenceClassification.from_pretrained(f"ronaldahmed/ccl_win-{model_type}")
100
  self._model.to(device)
101
  self._model.eval()
102