Spaces:
Sleeping
Sleeping
Update classicier.py
Browse files- classicier.py +2 -2
classicier.py
CHANGED
|
@@ -23,8 +23,8 @@ class Classicier(evaluate.Measurement):
|
|
| 23 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 24 |
|
| 25 |
# Load the tokenizer and model from the specified repository
|
| 26 |
-
self.tokenizer = AutoTokenizer.from_pretrained("
|
| 27 |
-
self.model = AutoModelForSequenceClassification.from_pretrained("
|
| 28 |
|
| 29 |
self.model.to(device)
|
| 30 |
self.device = device
|
|
|
|
| 23 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 24 |
|
| 25 |
# Load the tokenizer and model from the specified repository
|
| 26 |
+
self.tokenizer = AutoTokenizer.from_pretrained("Baleegh/Fluency_Score")
|
| 27 |
+
self.model = AutoModelForSequenceClassification.from_pretrained("Baleegh/Fluency_Score")
|
| 28 |
|
| 29 |
self.model.to(device)
|
| 30 |
self.device = device
|