StartupMap-India / backend /ml /__init__.py
Ram2005's picture
fix: Make ML __init__.py use lazy imports to avoid import errors
7c9766d verified
Raw
History Blame Contribute Delete
426 Bytes
# Bharat Tech Atlas — ML Integration Module
# Uses lazy imports to avoid crashes when optional deps (transformers, torch) aren't installed.
def get_classifier():
from .classifier import StartupSectorClassifier
return StartupSectorClassifier
def get_predictor():
from .predictor import GrowthPredictor
return GrowthPredictor
def get_server():
from .serving import ModelServer
return ModelServer