zayanomar5 commited on
Commit
9ed9a4e
·
verified ·
1 Parent(s): 82407c0

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +6 -3
main.py CHANGED
@@ -69,16 +69,18 @@ def compare():
69
  similarity_scores.append({"job": jobs_skills[i], "similarity_score": similarity_score.item()})
70
 
71
  return jsonify(similarity_scores)
72
- @app.route('/compare', methods=['POST'])
73
  def compare_jop():
74
  employee_skills = request.json.get('post')
75
  jobs_skills = request.json.get('employee_skills')
76
 
77
  if not isinstance(employee_skills, list) or not all(isinstance(skill, str) for skill in employee_skills):
 
 
78
  raise ValueError("employee_skills must be a list of strings")
79
 
80
- job_embeddings = model.encode(employee_skills)
81
- employee_embeddings = model.encode(emplpostoyee_skills)
82
 
83
  similarity_scores = []
84
  employee_embeddings_tensor = torch.from_numpy(employee_embeddings).unsqueeze(0)
@@ -91,5 +93,6 @@ def compare_jop():
91
  return jsonify(similarity_scores)
92
 
93
 
 
94
  if __name__ == '__main__':
95
  app.run()
 
69
  similarity_scores.append({"job": jobs_skills[i], "similarity_score": similarity_score.item()})
70
 
71
  return jsonify(similarity_scores)
72
+ @app.route('/compare_jop', methods=['POST'])
73
  def compare_jop():
74
  employee_skills = request.json.get('post')
75
  jobs_skills = request.json.get('employee_skills')
76
 
77
  if not isinstance(employee_skills, list) or not all(isinstance(skill, str) for skill in employee_skills):
78
+ raise ValueError("post must be a list of strings")
79
+ if not isinstance(jobs_skills, list) or not all(isinstance(skill, str) for skill in jobs_skills):
80
  raise ValueError("employee_skills must be a list of strings")
81
 
82
+ employee_embeddings = model.encode(employee_skills)
83
+ job_embeddings = model.encode(jobs_skills)
84
 
85
  similarity_scores = []
86
  employee_embeddings_tensor = torch.from_numpy(employee_embeddings).unsqueeze(0)
 
93
  return jsonify(similarity_scores)
94
 
95
 
96
+
97
  if __name__ == '__main__':
98
  app.run()