proyecto-final / app.py
lfernandopg's picture
Update app.py
cfe1092
raw
history blame
989 Bytes
import streamlit as st
from PIL import Image
from io import BytesIO
import tensorflow as tf
import matplotlib.pyplot as plt
import numpy as np
model = tf.keras.models.load_model('modelo.h5')
file = st.file_uploader("Por favor suba una imagen de ropa (jpg, png, jpeg)", type=['jpg','png','jpeg'])
if file is not None:
bytes_data = file.read()
x = np.array(Image.open(BytesIO(bytes_data)).convert('L'))
x = x / 255.00
x = np.expand_dims(x, axis=-1)
x = tf.image.resize(x, [80, 80])
plt.figure()
plt.imshow(x, cmap=plt.cm.binary)
plt.colorbar()
plt.grid(False)
plt.show()
x = np.repeat(x[:, :, np.newaxis], 3, axis=2)
x = np.squeeze(x)
col1, col2, col3 = st.columns(3)
with col1:
st.write("")
with col2:
image = Image.open(file)
st.markdown('<p style="text-align: center;">Image</p>',unsafe_allow_html=True)
st.image(image,width=300)
with col3:
st.write("")