Update app.py

app.py

@@ -6,6 +6,7 @@ from transformers import TrainingArguments, \
                          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
   ans = list(zip(probs,labels))