ImageTrust-AI / src /services /predictor.py
SiemonCha's picture
initial deployment
d581b00
raw
history blame contribute delete
364 Bytes
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