Add video

app.py

@@ -60,31 +60,51 @@ def load_lottieurl(url: str):
 
 def show_model_summary(expanded):
     st.subheader("Model gallery")
-    with st.expander('', expanded=expanded):
+    with st.expander('Image gallery', 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)
-            st.image('fauvism.png', width=200)
+            st.image('assets/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)
+            st.image('assets/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.image('assets/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)
+            st.image('assets/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.image('assets/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)
+            st.image('assets/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.image('assets/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)
+            st.image('assets/image/moon_gate.png', width=200)
+
+    with st.expander('Video gallery', expanded=False):
+        cols=st.columns(4)
+        cols[0].write("Universe GAN")
+        cols[0].image('assets/video/universe.gif')
+        cols[0].write("Fauvism still life GAN")
+        cols[0].image('assets/video/fauvism.gif')
+
+        cols[1].write("Aurora GAN")
+        cols[1].image('assets/video/aurora.gif')
+        cols[1].write("Moon gate GAN")
+        cols[1].image('assets/video/moon_gate.gif')
+
+        cols[2].write("Anime GAN")
+        cols[2].image('assets/video/anime.gif')
+        cols[2].write("Painting GAN")
+        cols[2].image('assets/video/painting.gif')
+
+        cols[3].write("Grumpy cat GAN")
+        cols[3].image('assets/video/grumpy_cat.gif')
 
 
 def main():
@@ -119,7 +139,7 @@ def main():
     <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>
+    <a href="https://www.linkedin.com/in/vumichien/" target="_blank">Chien Vu</a> | <a href="https://www.linkedin.com/in/nhu-hoang/" target="_blank">Nhu Hoang</a>
     <br/>
     </p>
             """,
@@ -135,7 +155,7 @@ def main():
         with col11:
             img_type = st.selectbox("Choose type of image to generate", index=0,
                                     options=["aurora", "anime", "painting", "fauvism", "shell", "universe", "grumpy cat", "moon gate"])
-                                    
+
             number_imgs = st.slider('How many images you want to generate ?', min_value=1, max_value=5)
             if number_imgs is None:
                 st.write('Invalid number ! Please insert number of images to generate !')
@@ -162,7 +182,7 @@ def main():
                         st.image(gan_images[1], width=300)
                     else:
                         st.image(gan_images[1:], width=150)
-                        
+
     elif choose == 'Mix style':
         st.header("Mix style")
         st.markdown(
@@ -190,7 +210,7 @@ def main():
                     mix_imgs = Image.fromarray(mix_imgs)
             with col22:
                 st.image(mix_imgs, width=600)
-                
+
 
 if __name__ == '__main__':
     main()