Update app.py

app.py

@@ -48,16 +48,21 @@ for uploaded_file in uploaded_files:
         entities = model.predict_entities(text_data, labels)
         df = pd.DataFrame(entities)
         st.dataframe(entities)
+        st.dataframe(df)
 
         import plotly.express as px
-        fig = px.treemap(df, path=[px.Constant("all"), 'text', 'label', 'score'],
-                 values='label', color='text')
+        fig = px.treemap(entities, path=[px.Constant("all"), 'text', 'label'],
+                values='score', color='label')
         fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
         st.plotly_chart(fig)
 
 
 
         
+
+
+
+        
             
         
         vectorizer = TfidfVectorizer()