joko333's picture
Load prediction model and update Streamlit app interface
e7d2b25
raw
history blame
343 Bytes
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')