Update app.py

app.py

@@ -3,6 +3,24 @@ from transformers import AutoTokenizer, AutoModelForSequenceClassification
 import torch
 import numpy as np
 
+MODEL_PATH = 'finiteautomata/bertweet-base-sentiment-analysis'
+tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
+model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
+model = model.to(device)
+
+
+logits = outputs.logits
+sigmoid = torch.nn.Sigmoid()
+probs = sigmoid(logits.squeeze().cpu())
+probs = probs.detach().numpy()
+
+for i, k in enumerate(label2id.keys()):
+    label2id[k] = probs[i]
+
+
+label2id = {k: v for k, v in sorted(label2id.items(), key=lambda item: item[1], reverse=True)}
+label2id
+
 tokenizer = AutoTokenizer.from_pretrained(MODEL_PATH)
 model = AutoModelForSequenceClassification.from_pretrained(MODEL_PATH)
 model = model.to(device)