Update app.py

app.py

@@ -100,11 +100,11 @@ def main():
 
     with st.sidebar:
         st_lottie(lottie_penguin, height=200)
-        choose = option_menu(None, ["Model Gallery", "Create your own images", "Style mix"],
-                             icons=['stack', 'file-plus', 'toggles'],
-                             menu_icon="cast", default_index=0,
+        choose = option_menu("FastGAN", ["Model Gallery", "Generate images", "Mix style"],
+                             icons=['collection', 'file-plus', 'intersect'],
+                             menu_icon="infinity", 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"},
                             }
                              )
@@ -112,19 +112,15 @@ def main():
         """
     ___
     <p style='text-align: center'>
-    FastGAN is an few-shot GAN model that generates images of several types!
+    FastGAN is a few-shot GAN model trained on high-fidelity images which requires less computation resource and samples for training.
+    <br/>
+    <a href="https://arxiv.org/abs/2101.04775" target="_blank">Article</a>
     </p>
-    <p style='text-align: center'>
-    Model training and Space creation by
+    <p style='text-align: center; font-size: 14px;'>
+    Model training and Spaces creating by
     <br/>
     <a href="https://huggingface.co/vumichien" target="_blank">Chien Vu</a> | <a href="https://huggingface.co/geninhu" target="_blank">Nhu Hoang</a>
     <br/>
-    </p>
-
-    <p style='text-align: center'>
-    based on
-    <br/>
-    <a href="https://github.com/silentz/Towards-Faster-And-Stabilized-GAN-Training-For-High-Fidelity-Few-Shot-Image-Synthesis" target="_blank">FastGAN model</a> | <a href="https://arxiv.org/abs/2101.04775" target="_blank">Article</a>
     </p>
             """,
         unsafe_allow_html=True,
@@ -133,7 +129,7 @@ def main():
     if choose == 'Model Gallery':
         st.header("Welcome to FastGAN")
         show_model_summary(True)
-    elif choose == 'Create your own images':
+    elif choose == 'Generate images':
         st.header("Generate images")
         col11, col12, col13 = st.columns([3,3.5,3.5])
         with col11:
@@ -167,8 +163,8 @@ def main():
                     else:
                         st.image(gan_images[1:], width=150)
                         
-    elif choose == 'Style mix':
-        st.header("Style mix")
+    elif choose == 'Mix style':
+        st.header("Mix style")
         st.markdown(
             """
         <p style='text-align: left'>