kanha-upadhyay commited on
Commit
6874c0e
Β·
1 Parent(s): a5247bb

Update spaCy model loading to use 'en_core_web_sm' instead of 'en_core_web_trf'

Browse files
Files changed (1) hide show
  1. src/utils/_model_manager.py +1 -1
src/utils/_model_manager.py CHANGED
@@ -42,7 +42,7 @@ class ModelManager:
42
  logger.info("βœ… Doctr model loaded successfully!")
43
 
44
  # Load spaCy model
45
- self._spacy_model = spacy.load(os.getenv("SPACY_MODEL_NAME", "en_core_web_trf"))
46
  logger.info(f"βœ… spaCy model loaded successfully!")
47
  self._models_loaded = True
48
  logger.info("πŸŽ‰ All models loaded successfully!")
 
42
  logger.info("βœ… Doctr model loaded successfully!")
43
 
44
  # Load spaCy model
45
+ self._spacy_model = spacy.load(os.getenv("SPACY_MODEL_NAME", "en_core_web_sm"))
46
  logger.info(f"βœ… spaCy model loaded successfully!")
47
  self._models_loaded = True
48
  logger.info("πŸŽ‰ All models loaded successfully!")