Spaces:
Sleeping
Sleeping
ronaldahmed
commited on
Commit
•
c2bcb6c
1
Parent(s):
7536bc5
Update ccl_win.py
Browse files- 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 |
|