app.py

import streamlit as st
import pickle

st.title("Salary prediction model based on experience, test score and interview score :heavy_dollar_sign:")
model = pickle.load(open("salary.pkl", "rb"))
experience = st.number_input("Experience:", 1,10)
test = st.number_input("Test Score:", 1,10)
interview = st.number_input("Interview Score:", 1,10)
if st.button("Predict"):
    prediction = model.predict([[experience, test, interview]])
    prediction = round(prediction[0][0], 2)
    st.success(f"Predicted Salary:{prediction}")