To reduce the number of man-hours required to inspect wires drawn through a die, we developed an AI tool that can predict the diameter of a wire after it is drawn using information on the shape and dimensions of the die. We then verified the prediction accuracy of this tool.