Update app.py

app.py

@@ -49,7 +49,12 @@ for uploaded_file in uploaded_files:
         df = pd.DataFrame(entities)
         st.dataframe(entities)
 
-        
+        import plotly.express as px
+        fig = px.treemap(df, path=[px.Constant("all"), 'Text', 'Label', 'Score'],
+                 values='Label', color='Text')
+        fig.update_layout(margin = dict(t=50, l=25, r=25, b=25))
+        st.plotly_chart(fig)
+