Spaces:
Build error
Build error
| from tensorflow.keras.models import load_model | |
| from PIL import Image | |
| import numpy as np | |
| import tensorflow as tf | |
| import streamlit as st | |
| model = load_model('AgeClassifier.h5') | |
| model2 = load_model('GenderClassifier.h5') | |
| # Open the video file | |
| f = st.file_uploader("Choose a Photo") | |
| # Read the video file from the file-like object | |
| if f is not None: | |
| img = Image.open(f) | |
| img2 = img.resize((200,200)) | |
| img = img.resize((256,256)) | |
| img = np.reshape(img,(1,256,256,3)) | |
| img2 = np.reshape(img2,(1,200,200,3)) | |
| pred,pred2 = model.predict(img2),model2.predict(img) | |
| st.image(img,use_column_width=True) | |
| st.write(np.argmax(pred)) | |
| st.write(pred) | |
| st.write(pred2) | |
| if pred2[0]>= 0.5: | |
| st.write('Male') | |
| else: | |
| st.write('Female') | |