Update app.py

app.py

@@ -59,7 +59,8 @@ def load_lottieurl(url: str):
     return r.json()
 
 def show_model_summary(expanded):
-    with st.expander("Model gallery", expanded=expanded):
+    st.subheader("Model gallery")
+    with st.expander('', expanded=expanded):
         col1, col2, col3, col4 = st.columns(4)
         with col1:
             st.markdown('Fauvism GAN [model](https://huggingface.co/huggan/fastgan-few-shot-fauvism-still-life)', unsafe_allow_html=True)
@@ -103,7 +104,7 @@ def main():
                              icons=['stack', 'file-plus'],
                              menu_icon="cast", default_index=0,
                              styles={
-                                "container": {"padding": ".0rem", "font-size": "14px", "background-color": "#11ffee00"},
+                                "container": {"padding": ".0rem", "font-size": "14px"},
                                 "nav-link-selected": {"color": "#000000", "font-size": "16px"},
                             }
                              )
@@ -143,7 +144,7 @@ def main():
         col11, col12, col13 = st.columns([3,3.5,3.5])
         with col11:
             img_type = st.selectbox("Choose type of image to generate", index=0,
-                                    options=["aurora", "anime", "painting", "fauvism", "shell", "universe", "grumpy cat"])
+                                    options=["aurora", "anime", "painting", "fauvism", "shell", "universe", "grumpy cat", "moon gate"])
         # with col12:
             number_imgs = st.slider('How many images you want to generate ?', min_value=1, max_value=5)
             if number_imgs is None: