Update app.py

app.py

@@ -39,9 +39,6 @@ config.id2label = {0: 'NEGATIVE', 1: 'NEUTRAL', 2: 'POSITIVE'}
 config.label2id = {"NEGATIVE": 0, "NEUTRAL": 1, "POSITIVE": 2}
 
 # creating a function used for gradio app
-# Creating dictionary
-dictionary = {}
-
 def sentiment_analysis(text):
 
     # Encode the text using the tokenizer
@@ -52,21 +49,13 @@ def sentiment_analysis(text):
     # Your code to get the scores for each class
     scores = output[0][0].detach().numpy()
     scores = softmax(scores)
-    
-    # Convert the numpy array into a list
-    scores = scores.tolist()
-    ranking = np.argsort(scores)
-    ranking = ranking[::-1]
-
-    for i in range(len(scores)):
-        l = config.id2label[ranking[i]]
-        s = scores[ranking[i]]
-       
-        # Convert the numpy float32 object into a float
-        dictionary[l] = float(s)
+
+    labels = {0: "NEGAITVE", 1: "NEUTRAL", 2: "POSITIVE"}
+    scores = {labels[i] float(s) for i , s in enumerate(scores) }
+
 
     # Return the dictionary as the response content
-    return dictionary
+    return scores
 
 # Create your interface
 demo = gr.Interface(