Update app.py

app.py

@@ -6,3 +6,7 @@ st.caption('***IMA GEN*** is a *text to image* generator that uses stable diffus
 prompt = st.text_input(':green[prompt.]')
 steps = st.slider(':green[Steps.]')
 generate = st.button(':green[Imagine]')
+
+
+if generate:
+    st.spinner(':red[pls wait..]')