Update app.py

app.py

@@ -41,22 +41,26 @@ config.label2id = {"NEGATIVE": 0, "NEUTRAL": 1, "POSITIVE": 2}
 # creating a function used for gradio app
 def sentiment_analysis(text):
 
+    # Create a new dictionary
+    scores = {}
+    
     # Encode the text using the tokenizer
     encoded_input = tokenizer(text, return_tensors='pt')
     # Get the output logits from the model
     output = model(**encoded_input)
-
+    
     # Your code to get the scores for each class
     scores = output[0][0].detach().numpy()
     scores = softmax(scores)
-
+    
     labels = {0: "NEGAITVE", 1: "NEUTRAL", 2: "POSITIVE"}
-    scores = {labels[i] float(s) for i , s in enumerate(scores) }
-
-
+    scores = {labels[i]: float(s) for i , s in enumerate(scores) }
+    
     # Return the dictionary as the response content
     return scores
 
+
+
 # Create your interface
 demo = gr.Interface(
     fn=sentiment_analysis,