import streamlit as st from psy import train_model x = st.slider('Select a value') st.write(x, 'squared is', x * x) psy.train_model()