Upload app.py

app.py

@@ -10,6 +10,7 @@ from torchvision.utils import make_grid
 from PIL import Image
 
 from streamlit_lottie import st_lottie
+from streamlit_option_menu import option_menu
 import requests
 
 device = 'cuda' if torch.cuda.is_available() else 'cpu'
@@ -98,6 +99,12 @@ def main():
 
     with st.sidebar:
         st_lottie(lottie_penguin, height=200)
+        choose = option_menu("FastGAN",
+                             ["Model Gallery", "Create your own images"],
+                             #icons=['house', 'camera fill'],
+                             menu_icon="app-indicator", default_index=0,
+                             styles={"container": {"border-radius": ".0rem"}}
+                             )
     st.sidebar.markdown(
         """
     ___
@@ -121,11 +128,16 @@ def main():
     )
 
     st.header("Welcome to FastGAN")
-    checked = st.checkbox('Click here if you want to create one of your own !')
 
-    if not checked:
+    if choose == 'Model Gallery':
         show_model_summary(True)
-    if checked:
+    elif choose == 'Create your own images':
+        #checked = st.checkbox('Click here if you want to create one of your own !')
+
+    # if not checked:
+    #     show_model_summary(True)
+    # if checked:
+        # show_model_summary(False)
         col11, col12, col13 = st.columns([3,3.5,3.5])
         with col11:
             img_type = st.selectbox("Choose type of image to generate", index=0,