Update app.py

app.py

@@ -11,7 +11,9 @@ def img_prep(paths):
     for i in paths:
         img = Image.open(i)
         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):
@@ -163,6 +165,7 @@ 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)