import streamlit as st
from PIL import Image
from tensorflow.keras.models import load_model
import numpy as np

model=load_model('citation_new.h5')
uploaded_files = st.file_uploader("Choo se a plan image file",type='jpg')
image = uploaded_files.read()


st.image(image, caption='your plan')

v1=st.sidebar.number_input('Insert a number')
v2=st.sidebar.number_input('Insert a number')
v3=st.sidebar.number_input('Insert a number')
t=np.array(image)
t=np.reshape(64,64,3)

predict=model.predict([[[v1,v2,v3],t]]
st.write(predict)