| 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}') | |