ui

app.py

@@ -104,9 +104,13 @@ button_stop = st.button('Stop')
 
 st.write("During generation, the two images show different ways of looking at the same process.\n- The left image starts with 100% noise and gradually turns into 100 signal.\n- The right image shows the model's current 'guess' about the left image will look like when all the noise has been removed.")
 
+generating_marker = st.empty()
+low_res = st.empty()
+high_res = st.empty()
+
 if button_go:
-    low_res = st.empty()
-    high_res = st.empty()
+    with generating_marker.container():
+        st.write('Generating...')
 
     count_low_res, count_high_res = 0, 0
     times_low, times_high = [], []
@@ -148,3 +152,6 @@ if button_go:
 
         if button_stop:
             break
+
+    with generating_marker.container():
+        st.write('')