FIFA2022 / predict.py
nyomanyudisdeveloper's picture
Upload 5 files
58f46f5 verified
import streamlit as st
import pandas as pd
import numpy as np
import pickle
with open('model_svr.pkl', 'rb') as file_6:
model = pickle.load(file_6)
def run():
with st.form("my_form"):
nama = st.text_input('masukan nama player', value='nama')
age = st.number_input('masukan usia player', min_value=15, max_value= 40, value=20)
height = st.slider('Height',50, 250, 170)
weight = st.number_input('Weight',50,100,70)
price = st.number_input('Price',0,1000000,10000, help="Harga Pemain dalam euro " )
# Every form must have a submit button.
st.write('-'*50)
attack = st.selectbox('Attacking Work Rate', {'Low','Medium','High'},index=1)
defense = st.radio('Defensive Work Rate', {'Low','Medium','High'},index=1)
st.markdown('---')
pace = st.number_input('Pace',0,100,50)
shoot = st.number_input('Shoot',0,100,50)
passing = st.number_input('Passing',0,100,50)
dribble = st.number_input('Dribble',0,100,50)
defend = st.number_input('Defend',0,100,50)
physicality = st.number_input('Physicality',0,100,50)
submitted = st.form_submit_button("Submit")
st.write("Outside the form")
data_inf = {
'Name': nama,
'Age' : age,
'Height' : height,
'Weight' : weight,
'Price' : price,
'AttackingWorkRate': attack,
'DefensiveWorkRate': defense,
'PaceTotal': pace,
'ShootingTotal': shoot,
'PassingTotal': passing,
'DribblingTotal': dribble,
'DefendingTotal': defend,
'PhysicalityTotal': physicality
}
data_inf = pd.DataFrame([data_inf])
if submitted:
result= model.predict(data_inf)
st.write(f'## Player Rating: {round(result[0])}')
st.balloons()
st.snow()
if __name__ == '__main__':
run()