Spaces:

ilyshi
/

shad_ml2_2

Runtime error

ilyshi commited on Mar 24, 2022

Commit

7ed1e45

•

1 Parent(s): 2421d35

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -4,8 +4,8 @@ import numpy as np
 from transformers import TrainingArguments, \
                          Trainer, AutoTokenizer, DataCollatorWithPadding, \
                          AutoModelForSequenceClassification
-categories = ['biology', 'computer science', 'economics', 'electrics', 'finance',
-               'math', 'physics', 'statistics']
 labels = [i for i in range(len(categories))]
 def print_probs(logits):
   probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100

 from transformers import TrainingArguments, \
                          Trainer, AutoTokenizer, DataCollatorWithPadding, \
                          AutoModelForSequenceClassification
+categories = ['Biology', 'Computer science', 'Economics', 'Electrics', 'Finance',
+               'Math', 'Physics', 'Statistics']
 labels = [i for i in range(len(categories))]
 def print_probs(logits):
   probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100