include("../models/baseline.jl")

model = Baseline()

function run_step(img, speed, command)
    img = convert.(Float32, img[:,:,1:3]) / 255
    img = reshape(img, (size(img)..., 1))
    img = KnetArray{Float32, 4}(img)

    speed = KnetArray{Float32}([speed])

    preds = model(img)
    throttle = preds[1,1]
    steer = preds[2,1]
    return return throttle, steer
end