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])