Spaces:
Runtime error
Runtime error
Update utils.py
Browse files
utils.py
CHANGED
|
@@ -32,7 +32,7 @@ def get_labels(text, model, tokenizer, count_labels=8):
|
|
| 32 |
return result_labels
|
| 33 |
result_labels.append(pair[1])
|
| 34 |
|
| 35 |
-
|
| 36 |
def load_model():
|
| 37 |
tokenizer = DistilBertTokenizerFast()
|
| 38 |
model = DistilBertForSequenceClassification()
|
|
|
|
| 32 |
return result_labels
|
| 33 |
result_labels.append(pair[1])
|
| 34 |
|
| 35 |
+
|
| 36 |
def load_model():
|
| 37 |
tokenizer = DistilBertTokenizerFast()
|
| 38 |
model = DistilBertForSequenceClassification()
|