Update app.py

app.py

@@ -13,7 +13,6 @@ def img_prep(paths):
         img = img.resize((256,256))
         img = img.convert('RGB')
         out.append(img)
-        st.write(np.shape(img))
     return out
 
 def model_out(img,model_path):
@@ -165,10 +164,23 @@ elif selected == 'Projects':
         img_options = {'img1':1, 'img2':2}
         img_selected_option = st.selectbox('Select an Image', img_options)
         if img_selected_option == 'img1':
-            st.write(np.shape(imgs[0]))
             st.image(model_out(imgs[0],'Portfolio Projects/FloodAreaSegmentationUnetPix2Pix.h5'),use_column_width=False)
         if img_selected_option == 'img2':
             st.image(model_out(imgs[1],'Portfolio Projects/FloodAreaSegmentationUnetPix2Pix.h5'),use_column_width=False)
+    if selected_option == 'Cyclegan':
+        imgs = img_prep(['unet1.jpg','unet2.jpg'])
+        col1,col2 = st.columns(2)
+        with col1:
+            st.image(imgs[0],use_column_width=False)
+        with col2:
+            st.image(imgs[1],use_column_width=False)
+
+        img_options = {'img1':1, 'img2':2}
+        img_selected_option = st.selectbox('Select an Image', img_options)
+        if img_selected_option == 'img1':
+            st.image(model_out(imgs[0],'Portfolio Projects/photo2ukiyo.h5'),use_column_width=False)
+        if img_selected_option == 'img2':
+            st.image(model_out(imgs[1],'Portfolio Projects/photo2ukiyo.h5'),use_column_width=False)
         
     st.write('You selected:', selected_option)
 else: