Ratinganime / app.py
nichaphat's picture
Update app.py
3aebe19
Raw
History Blame Contribute Delete
1.94 kB
import joblib
import pandas as pd
import streamlit as st #<-
model = joblib.load('model.joblib') #<-
unique_values = joblib.load('unique_values.joblib') #<-
unique_Genre = unique_values["Genre"]
unique_Certificate = unique_values["Certificate"]
unique_Director = unique_values["Director"]
def main(): #<- defined main
st.title("Rating Anime") #<- app name
with st.form("questionaire"): #<- input from user
Genre=st.selectbox('Genre',options=unique_Genre) #<-
Certificate=st.selectbox('Certificate',options=unique_Certificate) #<-
Director=st.selectbox('Director',options=unique_Director)
Votes = st.slider('Votes',min_value=0, max_value=5000000) #<-
Gross=st.slider('Gross',min_value=0.0,max_value=1000.0)#<-
Metascore=st.slider('Metascore',min_value=0.0 ,max_value=100.0) #<-
Year= st.number_input('Year',min_value=1900,max_value=2025) #<-
Runtime=st.slider('Runtime',min_value=0.0,max_value=200.0) #<-
# clicked==True only when the button is clicked
clicked = st.form_submit_button("Predict Rating")
if clicked:
result=model.predict(pd.DataFrame({"Votes": [Votes],
"Gross": [Gross],
"Metascore":[Metascore],
'Year':[Year],
'Runtime':[Runtime],
'Genre':[Genre],
'Certificate':[Certificate],
'Director':[Director]}))
# Show prediction
result= '8' if result[0]==8 else '7'
#print report
st.success("Your predictor rating movie is "+result)
# Run main()
#when import but dont wanna run main so..
if __name__== "__main__":
main()