Various embodiments of the present technology generally relate to solutions for integrating machine learning models into industrial automation environments. More specifically, embodiments of the present technology include systems and methods for implementing machine learning models within industrial control code to improve performance, increase productivity, and add capability to existing control programs. In an embodiment, a system comprises an interface component configured to display a graphical representation of a machine learning asset in an industrial automation environment, wherein the graphical representation includes a visual indicator representative of an output from the machine learning asset. The interface component is further configured to adjust the visual indicator based on the output from the machine learning asset. In addition, a process control component is configured to control an industrial process in the industrial automation environment based at least in part on the output from the machine learning asset.