Update app.py

app.py

@@ -162,7 +162,7 @@ elif selected == 'Projects':
     selected_option = st.selectbox('Select an option', options)
 
     if selected_option == 'Unet':
-        imgs = img_prep(['unet1','unet2'])
+        imgs = img_prep(['unet1.jpg','unet2.jpg'])
         col1,col2 = st.columns(2)
         with col1:
             st.image(imgs[0],use_column_width=False)
@@ -172,9 +172,9 @@ elif selected == 'Projects':
         img_options = {'img1':1, 'img2':2}
         img_selected_option = st.selectbox('Select an option', options)
         if img_selected_option == 'img1':
-            st.image(imgs[0],use_column_width=False)
+            st.image(model_out(imgs[0],'Portfolio Projects/FloodAreaSegmentationUnetPix2Pix.h5'),use_column_width=False)
         if img_selected_option == 'img2':
-            st.image(imgs[1],use_column_width=False)
+            st.image(model_out(imgs[1],'Portfolio Projects/FloodAreaSegmentationUnetPix2Pix.h5'),use_column_width=False)
         
     st.write('You selected:', selected_option)
 else: