Spaces:
Sleeping
Sleeping
File size: 364 Bytes
d581b00 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 | from src.models.inference import load_model, predict
# Load model once at startup, reuse for all predictions
_model = None
def get_model():
global _model
if _model is None:
_model = load_model()
return _model
def run_prediction(image_path: str) -> dict:
model = get_model()
result = predict(image_path, model=model)
return result |