Spaces:

Dzhamb
/

ArticleClassification

Runtime error

Dzhamb commited on Mar 26, 2022

Commit

783d3ba

1 Parent(s): 4261a4c

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -37,5 +37,5 @@ def get_labels(text, model, tokenizer, count_labels=8):
    tokenizer = DistilBertTokenizerFast()
    model = DistilBertForSequenceClassification()
    model.load_state_dict(torch.load('weight_model'))
-   return model

    tokenizer = DistilBertTokenizerFast()
    model = DistilBertForSequenceClassification()
    model.load_state_dict(torch.load('weight_model'))
+   return model, tokenizer