Update app.py

app.py

@@ -132,8 +132,9 @@ if st.session_state['upload_count'] < max_attempts:
             job_embedding = model.encode([job_desc])
             resume_embedding = model.encode([text_data])
             similarity_score = model.similarity(job_embedding, resume_embedding)[0][0]
+            similarity_array = np.array([[similarity_score]])
             st.subheader("Similarity between Applicant's Profile and Job Description", divider = "orange")
-            fig = px.imshow(similarity_score, text_auto=True,
+            fig = px.imshow(similarity_array, text_auto=True,
                                     labels=dict(x="Keyword similarity", y="Resumes", color="Productivity"),
                                     x=['Resume', 'Jon Description'],
                                     y=['Resume', 'Job Description'])