ui

app.py

@@ -93,6 +93,8 @@ gs1 = st.select_slider('Guidance scale (base)', [0.5*i for i in range(9)], value
 button_go = st.button('Generate')
 button_stop = st.button('Stop')
 
+st.write("During generation, the two images show different ways of looking at the same process.\n\nThe left image starts with 100% noise and gradually turns into 100 sigal.\n\nThe right image shows the model's current 'guess' about the left image will look like when all the noise has been removed.")
+
 if button_go:
     low_res = st.empty()
     high_res = st.empty()
@@ -119,6 +121,7 @@ if button_go:
             total = ts2
             times_high.append(delta)
             times = times_high
+            prefix = "Part 1 of 2 (base model)"
         else:
             target = low_res
             count_low_res += 1
@@ -126,12 +129,13 @@ if button_go:
             total = ts1
             times_low.append(delta)
             times = times_low
+            prefix = "Part 2 of 2 (upsampling)"
 
         rate = sum(times)/len(times)
 
         with target.container():
             st.image([s, xs])
-            st.write(f'{count} / {total} | {rate:.2f} seconds/frame')
+            st.write(f'{prefix} | {count:02d} / {total} frames | {rate:.2f} seconds/frame')
 
         if button_stop:
             break