mrfirdauss commited on
Commit
ffe7632
·
verified ·
1 Parent(s): 2be094f

Upload folder using huggingface_hub

Browse files
Files changed (2) hide show
  1. kafkaWorker.py +2 -0
  2. requirements.txt +2 -1
kafkaWorker.py CHANGED
@@ -56,6 +56,8 @@ def send_results_back(full_results: dict[str, any], job_application_id: str):
56
  "job_application_id": job_application_id,
57
  **full_results
58
  }
 
 
59
  print(f"Sending data: {body}")
60
  response = requests.patch(url, json=body, headers=headers)
61
  print(f"Data sent with status code {response.status_code}")
 
56
  "job_application_id": job_application_id,
57
  **full_results
58
  }
59
+ body['score'] = body['score'].tolist()[0]
60
+
61
  print(f"Sending data: {body}")
62
  response = requests.patch(url, json=body, headers=headers)
63
  print(f"Data sent with status code {response.status_code}")
requirements.txt CHANGED
@@ -4,4 +4,5 @@ sentence-transformers
4
  scikit-learn
5
  python-dotenv
6
  kafka-python
7
- pandas
 
 
4
  scikit-learn
5
  python-dotenv
6
  kafka-python
7
+ pandas
8
+ numpy