Update app.py

app.py

@@ -24,6 +24,7 @@ model_name = {
     "universe": 'huggan/fastgan-few-shot-universe',
     "grumpy cat": 'huggan/fastgan-few-shot-grumpy-cat',
     "anime": 'huggan/fastgan-few-shot-anime-face',
+    "moon gate": 'huggan/fastgan-few-shot-moongate',
 }
 
 #@st.cache(allow_output_mutation=True)
@@ -63,27 +64,26 @@ def show_model_summary(expanded):
         with col1:
             st.markdown('Fauvism GAN [model](https://huggingface.co/huggan/fastgan-few-shot-fauvism-still-life)', unsafe_allow_html=True)
             st.image('fauvism.png', width=200)
-
             st.markdown('Painting GAN [model](https://huggingface.co/huggan/fastgan-few-shot-painting)', unsafe_allow_html=True)
             st.image('painting.png', width=200)
 
         with col2:
             st.markdown('Aurora GAN [model](https://huggingface.co/huggan/fastgan-few-shot-aurora)', unsafe_allow_html=True)
             st.image('aurora.png', width=200)
-
             st.markdown('Universe GAN [model](https://huggingface.co/huggan/fastgan-few-shot-universe)', unsafe_allow_html=True)
             st.image('universe.png', width=200)
 
         with col3:
             st.markdown('Anime GAN [model](https://huggingface.co/huggan/fastgan-few-shot-anime-face)', unsafe_allow_html=True)
             st.image('anime.png', width=200)
-
             st.markdown('Shell GAN [model](https://huggingface.co/huggan/fastgan-few-shot-shells)', unsafe_allow_html=True)
             st.image('shell.png', width=200)
 
         with col4:
             st.markdown('Grumpy cat GAN [model](https://huggingface.co/huggan/fastgan-few-shot-grumpy-cat)', unsafe_allow_html=True)
             st.image('grumpy_cat.png', width=200)
+            st.markdown('Moon gate GAN [model](https://huggingface.co/huggan/fastgan-few-shot-moongate)', unsafe_allow_html=True)
+            st.image('moon_gate.png', width=200)
 
 
 def main():
@@ -100,9 +100,12 @@ def main():
     with st.sidebar:
         st_lottie(lottie_penguin, height=200)
         choose = option_menu(None, ["Model Gallery", "Create your own images"],
-                             #icons=['house', 'camera fill'],
-                             menu_icon="app-indicator", default_index=0,
-                             styles={"container": {"border-radius": ".0rem"}}
+                             icons=['stack', 'file-plus'],
+                             menu_icon="cast", default_index=0,
+                             styles={
+                                "container": {"padding": ".0rem", "font-size": "14px", "background-color": "#11ffee00"},
+                                "nav-link-selected": {"color": "#000000", "font-size": "16px"},
+                            }
                              )
     st.sidebar.markdown(
         """
@@ -155,10 +158,11 @@ def main():
                 unsafe_allow_html=True,)
 
         if generate_button:
-            with st.spinner(text=f"Loading selected model..."):
-                generator = load_generator(model_name[img_type])
-            with st.spinner(text=f"Generating images..."):
-                gan_images = generate_images(generator, number_imgs)
+            with col11:
+                with st.spinner(text=f"Loading selected model..."):
+                    generator = load_generator(model_name[img_type])
+                with st.spinner(text=f"Generating images..."):
+                    gan_images = generate_images(generator, number_imgs)
             with col12:
                 st.image(gan_images[0], width=300)
             if len(gan_images) > 1: