INNER CODE UNIT · Python

predict

DataTalksClub/machine-learning-zoomcamp · cohorts/2026/10-kubernetes/workshop/service/app.py:56

def predict(url: str):
    X = preprocessor.from_url(url)
    result = session.run([output_name], {input_name: X})
    float_predictions = result[0][0].tolist()
    predictions_dict = dict(zip(classes, float_predictions))
    
    top_class = max(predictions_dict, key=predictions_dict.get)
    top_probability = predictions_dict[top_class]
    
    return predictions_dict, top_class, top_probability


@app.get("/")
def root():
    return {"message": "Clothing Classification Service"}


@app.get("/health")

View source record →