Update app.py

app.py

@@ -128,6 +128,7 @@ if st.session_state['upload_count'] < max_attempts:
             labels = ["person", "country", "organization", "role", "skills"]
             entities = model.predict_entities(text_data, labels)
             df = pd.DataFrame(entities)
+            st.subheader("Applicant's Profile", divider = "orange")
             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))
@@ -137,6 +138,7 @@ if st.session_state['upload_count'] < max_attempts:
             tfidf_df = pd.DataFrame(tfidf_matrix.toarray(), columns=vectorizer.get_feature_names_out())
             cosine_sim_matrix = cosine_similarity(tfidf_matrix)
             cosine_sim_df = pd.DataFrame(cosine_sim_matrix)
+            st.subheader("Similarity between Applicant's Profile and job description", divider = "orange")
 
             fig = px.imshow(cosine_sim_df, text_auto=True,
                             labels=dict(x="Keyword similarity", y="Resumes", color="Productivity"),