I want t test on my robot arm, I dont know the input/output for these models, could you release inference code