Update main.py

main.py

@@ -62,7 +62,8 @@ def compare():
     for i, job_e in enumerate(job_embeddings):
         job_e_tensor = torch.from_numpy(job_e).unsqueeze(0)
         similarity_score = torch.nn.functional.cosine_similarity(employee_embeddings_tensor, job_e_tensor, dim=1)
-        similarity_scores.append({"job": jobs_skills[i], "similarity_score": similarity_score.item()})
+        similarity_scores.append({"job": jobs_skills[i], "similarity_score": similarity_score.squeeze().tolist()})
+
 
     return jsonify(similarity_scores)