import streamlit as st
import pickle

st.title('Salary estimate based on experience, written and interview :heavy_dollar_sign:')
model = pickle.load(open('src/salary.pkl', 'rb'))
experience = st.number_input('experience', 1, 10)
written = st.number_input('written', 1, 10)
interview = st.number_input('interview', 1, 10)

if st.button('Guess it'):
    guess = model.predict([[experience, written, interview]])
    guess = round(guess[0])
    st.success(f'Your estimated salary is: {guess}')