Update app.py

app.py

@@ -191,7 +191,7 @@ elif selected == 'Projects':
             st.image(model_out(imgs[1],'Portfolio Projects/photo2monet2.h5'),use_column_width=False)
 
     if selected_option == 'Pix2Pix':
-        imgs = img_prep(['pix2pix1.jpg','pix2pix2.jpg'])
+        imgs = img_prep(['pp1.jpg','pp2.jpg'])
         col1,col2 = st.columns(2)
         with col1:
             st.image(imgs[0],use_column_width=False)
@@ -201,9 +201,26 @@ 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.image(model_out(imgs[0],'Portfolio Projects/Aerialtogooglemaps.h5'),use_column_width=False)
+            st.image(model_out(imgs[0],'Portfolio Projects/FaceWithMask.h5'),use_column_width=False)
         if img_selected_option == 'img2':
-            st.image(model_out(imgs[1],'Portfolio Projects/Aerialtogooglemaps.h5'),use_column_width=False)
+            st.image(model_out(imgs[1],'Portfolio Projects/FaceWithMask.h5'),use_column_width=False)
+
+    if selected_option == 'DCGAN':
+        model = tf.keras.models.load_model('doggen3.h5')
+
+# Create a button
+        button_clicked = st.button("Generate")
+
+# Check if the button is clicked
+        if button_clicked:
+    # Generate an image
+            seed = tf.random.normal((1, 100))
+            pred = model.predict(seed)
+            pred = pred * 0.5 + 0.5  # Normalize the pixel values
+            pred = np.squeeze(pred)  # Remove singleton dimensions if any
+    
+    # Display the generated image
+    st.image(pred, caption='Generated Image', use_column_width=True)
         
     st.write('You selected:', selected_option)
 else: