actually draw

app.py

@@ -22,7 +22,9 @@ DIFFUSION_DEFAULTS = dict(
 @st.experimental_singleton
 def setup():
     import os, subprocess, sys
-    os.system("git clone https://github.com/nostalgebraist/improved-diffusion.git && cd improved-diffusion && git fetch origin nbar-space && git checkout nbar-space && pip install -e .")
+    if not os.path.exists('improved_diffusion'):
+        os.system("git clone https://github.com/nostalgebraist/improved-diffusion.git")
+    os.system("cd improved-diffusion && git fetch origin nbar-space && git checkout nbar-space && pip install -e .")
     os.system("pip install tokenizers x-transformers==0.22.0 axial-positional-embedding")
     os.system("pip install einops==0.3.2")
     sys.path.append("improved-diffusion")
@@ -78,8 +80,13 @@ def handler(text, ts1, ts2, gs1):
 text = st.text_area('asdf')
 
 if st.button('rweerew'):
+    low_res = st.empty()
+    high_res = st.empty()
+
     for s, xs in handler(text, 20, 20, 0):
-        pass
+        target = low_res if s.size[0] < 256 else high_res
+        with target.container():
+            st.write([s, xs])
 
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)