updated app

app.py

@@ -89,5 +89,5 @@ if text and button:
     labels = ['Negative','Neutral','Positive']
     scores = {l:float(s) for (l,s) in zip(labels,scores_)}
     st.write("Logits: ",output.logits)
-    # y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
+    y_pred = np.argmax(output.logits.detach().numpy(),axis =1)
     # st.write("Prediction :",scores)