Spaces:
Sleeping
Sleeping
| import tensorflow as tf | |
| from PIL import Image | |
| import numpy as np | |
| import streamlit as st | |
| img = st.file_uploader("Choose a file") | |
| model = tf.keras.models.load_model('RockPaperScissor.h5') | |
| if img is not None: | |
| img = Image.open(img) | |
| img = img.resize((256,256)) | |
| img = np.reshape(img,(1,256,256,3)) | |
| pred = model.predict(img) | |
| st.write(pred) | |
| pred = np.argmax(pred) | |
| if pred == 0: | |
| st.write('PAPER') | |
| elif pred == 1: | |
| st.write('ROCK') | |
| else: | |
| st.write('SCISSOR') | |