evacuationTime / app.py
arnikdehnavi's picture
Update app.py
8757d74
raw
history blame contribute delete
527 Bytes
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)