ronaldahmed commited on
Commit
6314abb
1 Parent(s): 46feb6e

Update ccl_win.py

Browse files
Files changed (1) hide show
  1. 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("ronaldahmed/ccl_win/" + model_type,repo_type="space")
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