Update app.py

app.py

@@ -144,8 +144,10 @@ def main():
                 unsafe_allow_html=True,)
 
         if generate_button:
-            generator = load_generator(model_name[img_type])
-            gan_images = generate_images(generator, number_imgs)
+            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: