import streamlit as st from utils.prediction import load_model_for_prediction #import transformers #from transformers import pipeline st.title('My first app') x = st.slider('Select a value') st.write(x, 'squared is', x * x) # load the model and cache it model, label_encoder, tokenizer = load_model_for_prediction() st.write('Model loaded')