Upload 3 files

README.md

@@ -0,0 +1 @@
+# keras-streamlit

g_model_AtoB_002160.h5

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:af168a96d6f548ae084a9608211822e7da935f9ace3192d25a9eb77372acd1e8
+size 141314344

main.py

@@ -0,0 +1,39 @@
+
+import streamlit as st
+
+from tensorflow.keras.models import load_model
+
+import numpy as np
+
+import matplitlib.pyplot as plt
+
+st.header("Photo to Monet")
+
+st.caption('Upload an image 256x256')
+
+model = load_model('g_model_AtoB_002160.h5')
+
+@st.cache
+def load_image(image_file):
+    img=plt.imread(image_file)
+    return img
+
+imgpath = st.file_uploader("Choose a file", type =['png', 'jpeg', 'jpg'])
+
+if imgpath is not None:
+
+    img = load_image(imgpath )
+
+    st.image(img, width=250)
+
+
+def convert(image):
+    img=load_image(img,target_size=(256,256))
+    img_array = np.reshape(img, (1, 256, 256, 3))
+    result=model.predict(img_array)
+    result=np.squeeze(img,axis=0)
+    return result
+
+if st.button('Convert'):
+    result=convert(imagepath)
+    st.image(result)