Update app.py

app.py

@@ -23,6 +23,7 @@ def img_prep(paths):
         img = img.resize((256,256))
         img = ((np.asarray(img))/127.5)-1
         out.append(img)
+    return out
 
 def model_out(img,model_path):
     img = np.expand_dims(img,0)
@@ -163,7 +164,6 @@ elif selected == 'Projects':
 
     if selected_option == 'Unet':
         imgs = img_prep(['unet1.jpg','unet2.jpg'])
-        st.write(np.shape(imgs))
         col1,col2 = st.columns(2)
         with col1:
             st.image(imgs[0],use_column_width=False)