Update app.py

app.py

@@ -57,6 +57,19 @@ for uploaded_file in uploaded_files:
     
         data = {"Text": text_data, **entity_dict}
         st.dataframe(data)
+
+
+        if data is not None:
+            value_counts1 = data['label'].value_counts()
+                
+            df1 = pd.DataFrame(value_counts1)
+                
+            final_df = df1.reset_index().rename(columns={"index": "label"})
+                
+            
+          
+            fig2 = px.bar(final_df, x="count", y="label", color="label", text_auto=True, title='Occurrences of predicted labels')
+            st.plotly_chart(fig2)
             
         
         vectorizer = TfidfVectorizer()
@@ -73,8 +86,8 @@ for uploaded_file in uploaded_files:
         import plotly.express as px
         
         fig = px.imshow(cosine_sim_df, text_auto=True, labels=dict(x="Cosine similarity", y="Text", color="Productivity"),
-                        x=['text1', 'text2'],
-                        y=['text1', 'text2'])
+                        x=['text1', 'Jon Description'],
+                        y=['text1', 'Job Description'])
         st.plotly_chart(fig)
 
         st.subheader("Cosine Similarity Scores (Job Description vs. Resumes):")