Spaces:
Sleeping
Sleeping
Update ccl_win.py
Browse files- ccl_win.py +1 -3
ccl_win.py
CHANGED
|
@@ -61,8 +61,6 @@ Examples:
|
|
| 61 |
{'accuracy': 1.0}
|
| 62 |
"""
|
| 63 |
|
| 64 |
-
# TODO: Define external resources urls if needed
|
| 65 |
-
BAD_WORDS_URL = "http://url/to/external/resource/bad_words.txt"
|
| 66 |
WINDOW_SIZE = 3
|
| 67 |
|
| 68 |
|
|
@@ -98,7 +96,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(
|
| 102 |
self._model.to(device)
|
| 103 |
self._model.eval()
|
| 104 |
|
|
|
|
| 61 |
{'accuracy': 1.0}
|
| 62 |
"""
|
| 63 |
|
|
|
|
|
|
|
| 64 |
WINDOW_SIZE = 3
|
| 65 |
|
| 66 |
|
|
|
|
| 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 |
|