Update app.py

app.py

@@ -18,7 +18,10 @@ if st.button('Submit'):
     output = classifier(text_input)
 
     best_prediction = output[0]
+    sentiment = best_prediction['label']
+    confidence = best_prediction['score']
     
     # Display the sentiment prediction to the user
-    st.write(f'Sentiment: {best_prediction['label']}')
-    st.write(f'Confidence: {best_prediction['score']}')
+    st.write(f'Sentiment: {sentiment}')
+    st.write(f'Confidence: {confidence}')
+