Spaces:

Dzhamb
/

ArticleClassification

Runtime error

Dzhamb commited on Mar 26, 2022

Commit

627c1c5

1 Parent(s): 99ca19b

Update utils.py

Files changed (1) hide show

utils.py CHANGED Viewed

@@ -14,7 +14,7 @@ def get_text(title: str, abstract: str):
   return text
 def get_labels(text, model, tokenizer, count_labels=8):
-  tokens = tokeinizer(text, return_tensors='pt')
   outputs = model(**tokens)
   probs = torch.nn.Softmax(dim=count_labels)(outputs.logits)

   return text
 def get_labels(text, model, tokenizer, count_labels=8):
+  tokens = tokenizer(text, return_tensors='pt')
   outputs = model(**tokens)
   probs = torch.nn.Softmax(dim=count_labels)(outputs.logits)