prediction.py

import streamlit as st
import pickle

@st.cache_resource
def predict(data):
  model = pickle.load(open('model.pkl', 'rb'))
  return model.predict(data)