Spaces:
Runtime error
Runtime error
update
Browse files
app.py
CHANGED
@@ -4,6 +4,8 @@ import numpy as np
|
|
4 |
from transformers import TrainingArguments, \
|
5 |
Trainer, AutoTokenizer, DataCollatorWithPadding, \
|
6 |
AutoModelForSequenceClassification
|
|
|
|
|
7 |
def print_probs(logits):
|
8 |
probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100
|
9 |
ans = list(zip(probs,labels))
|
|
|
4 |
from transformers import TrainingArguments, \
|
5 |
Trainer, AutoTokenizer, DataCollatorWithPadding, \
|
6 |
AutoModelForSequenceClassification
|
7 |
+
categories = ['biology', 'computer science', 'economics', 'electrics', 'finance',
|
8 |
+
'math', 'physics', 'statistics']
|
9 |
def print_probs(logits):
|
10 |
probs = torch.nn.functional.softmax(logits, dim=0).numpy()*100
|
11 |
ans = list(zip(probs,labels))
|