youssefreda9 commited on
Commit
c6e2341
·
1 Parent(s): 4718622

fix: Always load models on import (RENDER/GUNICORN_LOADED env vars not set in HF Spaces)

Browse files
Files changed (1) hide show
  1. src/app.py +2 -3
src/app.py CHANGED
@@ -855,9 +855,8 @@ def _ensure_models_loaded():
855
  if not load_models():
856
  logger.error("Failed to load any models. Server will start but functionality will be limited.")
857
 
858
- # Load models on import (gunicorn) guarded by flag to prevent double-load
859
- if os.environ.get('RENDER') or os.environ.get('GUNICORN_LOADED'):
860
- _ensure_models_loaded()
861
 
862
 
863
  if __name__ == '__main__':
 
855
  if not load_models():
856
  logger.error("Failed to load any models. Server will start but functionality will be limited.")
857
 
858
+ # Load models on import (gunicorn imports this module, __name__ != '__main__')
859
+ _ensure_models_loaded()
 
860
 
861
 
862
  if __name__ == '__main__':