NOT-OMEGA commited on
Commit
a075d7d
·
verified ·
1 Parent(s): de30f06

Update processor_bert.py

Browse files
Files changed (1) hide show
  1. processor_bert.py +2 -5
processor_bert.py CHANGED
@@ -32,12 +32,10 @@ def preload_models():
32
  """Lazily load models — thread-safe, strict single initialization."""
33
  global _classifier, _model_ready
34
 
35
- # 🚨 GOOGLE-LEVEL FIX: Everything critical must be INSIDE the lock
36
  with _load_lock:
37
  if _classifier is None:
38
- logger.info("Initializing BERT pipeline...")
39
 
40
- # Initialization logic goes HERE, not outside
41
  _classifier = pipeline(
42
  task="text-classification",
43
  model=MODEL_NAME,
@@ -45,8 +43,7 @@ def preload_models():
45
  top_k=1
46
  )
47
  _model_ready = True
48
- logger.info("BERT pipeline ready.")
49
-
50
  def _load_models():
51
  """Lazily load models — thread-safe, sirf ek baar load hoga."""
52
  global _USE_ONNX, _embedding_model, _classifier, _ort_session, _ort_tokenizer, _model_ready
 
32
  """Lazily load models — thread-safe, strict single initialization."""
33
  global _classifier, _model_ready
34
 
 
35
  with _load_lock:
36
  if _classifier is None:
37
+ print("Initializing BERT pipeline...") # Yahan change kiya
38
 
 
39
  _classifier = pipeline(
40
  task="text-classification",
41
  model=MODEL_NAME,
 
43
  top_k=1
44
  )
45
  _model_ready = True
46
+ print("BERT pipeline ready.") # Yahan change kiya
 
47
  def _load_models():
48
  """Lazily load models — thread-safe, sirf ek baar load hoga."""
49
  global _USE_ONNX, _embedding_model, _classifier, _ort_session, _ort_tokenizer, _model_ready