Avinashnalla7 commited on
Commit
54801eb
·
1 Parent(s): 6c3783f

Log build revision at startup

Browse files
Files changed (1) hide show
  1. backend/worker/worker.py +2 -1
backend/worker/worker.py CHANGED
@@ -2,6 +2,7 @@ from __future__ import annotations
2
 
3
  import os
4
  import requests
 
5
 
6
  def _upload_pdf_to_api(pdf_id, pdf_path, pdf_name):
7
  base = (os.environ.get("PDF_TRAINER_API_BASE") or "").strip()
@@ -216,7 +217,7 @@ def main():
216
  tmp_dir = root / "tmp"
217
  tmp_dir.mkdir(parents=True, exist_ok=True)
218
 
219
- print(f"[worker] Watching label: {s.label_incoming}")
220
  print(f"[worker] Known label: {s.label_known}")
221
  print(f"[worker] Unknown label: {s.label_unknown}")
222
  print(f"[worker] Train label: {s.label_train}")
 
2
 
3
  import os
4
  import requests
5
+ GIT_SHA=os.getenv('HF_REVISION','') or os.getenv('GIT_SHA','')
6
 
7
  def _upload_pdf_to_api(pdf_id, pdf_path, pdf_name):
8
  base = (os.environ.get("PDF_TRAINER_API_BASE") or "").strip()
 
217
  tmp_dir = root / "tmp"
218
  tmp_dir.mkdir(parents=True, exist_ok=True)
219
 
220
+ print(f"[worker] build={GIT_SHA}\n[worker] Watching label: {s.label_incoming}")
221
  print(f"[worker] Known label: {s.label_known}")
222
  print(f"[worker] Unknown label: {s.label_unknown}")
223
  print(f"[worker] Train label: {s.label_train}")