Update app.py

app.py

@@ -1,3 +1,40 @@
+Hugging Face's logo
+Hugging Face
+Search models, datasets, users...
+Models
+Datasets
+Spaces
+Docs
+Solutions
+Pricing
+
+
+
+Hugging Face is way more fun with friends and colleagues! 🤗 Join an organization
+Spaces:
+
+jeromechew15
+/
+milestone3 Copied
+like
+0
+App
+Files
+Community
+milestone3
+/
+app.py
+jeromechew15's picture
+jeromechew15
+Update app.py
+0777fd8
+2 days ago
+raw
+history
+blame
+contribute
+delete
+3.37 kB
 import streamlit as st
 import pandas as pd
 import numpy as np
@@ -86,4 +123,9 @@ if st.button('Add to Results'):
         st.write('Tweet:', text_input)
         st.write('Highest Toxicity Class:', class_label)
         st.write('Probability:', class_prob)
-        st.session_state.classification_results = st.session_state.classification_results.append
+        st.session_state.classification_results = st.session_state.classification_results.append({
+            'tweet': text_input,
+            'toxicity_class': class_label,
+            'probability': class_prob
+        }, ignore_index=True)
+st.write(st.session_state.classification_results)