inference.py

import pandas as pd
import joblib

def predict_volume(input_data):
    model = joblib.load("traffic_volume_model.joblib")
    max_volume = 7280
    prediction = model.predict(input_data) * max_volume
    return prediction