date_type_clf / app.py
adil9858's picture
Upload 5 files
6f610d0 verified
rev_map = {0: 'BERHI',
1: 'DEGLET',
2: 'DOKOL',
3: 'IRAQI',
4: 'ROTANA',
5: 'SAFAVI',
6: 'SOGAY'}
import streamlit as st
import joblib
st.title('Dates Type Classification App')
st.write('This app takes in the features and classifies the date')
st.image('pic.jpg')
model = joblib.load('log_model.pkl')
scaler = joblib.load('scaler.pkl')
#['AREA', 'PERIMETER', 'MINOR_AXIS', 'CONVEX_AREA', 'ROUNDNESS', 'MeanRR',
# 'SkewRR', 'SkewRB', 'EntropyRG', 'ALLdaub4RR']
area = st.number_input('Enter area:')
perimeter = st.number_input('Enter perimeter:')
minor_axis = st.number_input('Enter minor_axis:')
convex_area = st.number_input('Enter convex_area:')
roundness = st.number_input('Enter roundness:')
meanrr = st.number_input('Enter meanrr:')
skewrr = st.number_input('Enter skewrr:')
skewrb = st.number_input('Enter skewrb:')
entropyrg = st.number_input('Enter entropyrg:')
ALLdaub4RR = st.number_input('Enter ALLdaub4RR:')
if st.button('Predict:--->'):
sample = [[area,perimeter,minor_axis,convex_area,roundness,meanrr,skewrr,skewrb,entropyrg,ALLdaub4RR]]
sample_scaled = scaler.transform(sample)
pred = model.predict(sample_scaled)
st.write(f"The class is : {rev_map[pred[0]]}")