prediction.py

import joblib
import streamlit as st
import numpy as np
def predict(data):
    clf=joblib.load("rf_model.sav")
    return clf.predict(data)
from predictions import predict
if st.button("Predict type of Iris"):
    result = predict(np.array([['Normal_Weight', 'Overweight_Level_I', 'Overweight_Level_II',
       'Obesity_Type_I', 'Insufficient_Weight', 'Obesity_Type_II',
       'Obesity_Type_III']]))
    st.text(result[0])