Abdulmohsena commited on
Commit
3919100
·
verified ·
1 Parent(s): bdcadb4

Update classicier.py

Browse files
Files changed (1) hide show
  1. 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("AbdulmohsenA/classicier")
27
- self.model = AutoModelForSequenceClassification.from_pretrained("AbdulmohsenA/classicier")
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