stop button

app.py

@@ -77,6 +77,7 @@ def handler(text, ts1, ts2, gs1):
 
     return pipeline.sample(**args)
 
+st.title('nostalgebraist-autoresponder image generation demo')
 
 text = st.text_area('Enter your text here (or leave blank for a textless image)', max_chars=380)
 
@@ -89,7 +90,10 @@ ts2 = st.slider('Steps (upsampling)', min_value=5, max_value=500, value=10, help
 gs1 = st.select_slider('Guidance scale (base)', [0.5*i for i in range(9)], value=0., help=help_gs1)
 
 
-if st.button('Generate'):
+button_go = st.button('Generate')
+button_stop = st.button('Stop')
+
+if button_go:
     low_res = st.empty()
     high_res = st.empty()
 
@@ -128,3 +132,6 @@ if st.button('Generate'):
         with target.container():
             st.image([s, xs])
             st.write(f'{count} / {total} | {rate:.2f} seconds/frame')
+
+        if button_stop:
+            break