Spaces:

ilyshi
/

shad_ml2_2

Runtime error

ilyshi commited on Mar 24, 2022

Commit

edcde04

•

1 Parent(s): 5aaeb9a

update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,6 +4,8 @@ import numpy as np
 from transformers import TrainingArguments, \
                          Trainer, AutoTokenizer, DataCollatorWithPadding, \
                          AutoModelForSequenceClassification
 def print_probs(logits):
   probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100
   ans = list(zip(probs,labels))

 from transformers import TrainingArguments, \
                          Trainer, AutoTokenizer, DataCollatorWithPadding, \
                          AutoModelForSequenceClassification
+categories = ['biology', 'computer science', 'economics', 'electrics', 'finance',
+               'math', 'physics', 'statistics']
 def print_probs(logits):
   probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100
   ans = list(zip(probs,labels))