File size: 491 Bytes
3571bcc a9de861 33c5484 3571bcc 52d2e6a 3571bcc 33c5484 3571bcc a9de861 3571bcc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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}')
|