ui

app.py

@@ -62,10 +62,10 @@ def setup():
     pipeline = improved_diffusion.pipeline.SamplingPipeline(sampling_model_sres1, sampling_model_sres2)
     return pipeline
 
+
 def handler(text, ts1, ts2, gs1):
     pipeline = setup()
 
-    # a = np.random.randint(0, 255, (128, 128, 3)).astype(np.uint8)
     data = {'text': text[:380], 'guidance_scale': gs1}
     args = {k: v for k, v in DIFFUSION_DEFAULTS.items()}
     args.update(data)
@@ -80,16 +80,16 @@ def handler(text, ts1, ts2, gs1):
 
 text = st.text_area('Enter your text here (or leave blank for a textless image)', max_chars=380)
 
-help_ts1 = "foo"
-help_ts2 = "bar " * 40
-help_gs1 = "aaff"
+help_ts1 = "How long to run the base model. Larger values make the image more realistic / better. Smaller values are faster."
+help_ts2 = "How long to run the upsampling model. Larger values sometimes make the big image crisper and more detailed. Smaller values are faster."
+help_gs1 = "Guidance scale. Larger values make the image more likely to contain the text you wrote. If this is zero, the first part will be faster."
 
 ts1 = st.slider('Steps (base)', min_value=5, max_value=500, value=10, help=help_ts1)
 ts2 = st.slider('Steps (upsampling)', min_value=5, max_value=500, value=10, help=help_ts2)
 gs1 = st.select_slider('Guidance scale (base)', [0.5*i for i in range(9)], value=0., help=help_gs1)
-# ts1, ts2, gs1 = 20, 20, 0
 
-if st.button('rweerew'):
+
+if st.button('Generate'):
     low_res = st.empty()
     high_res = st.empty()
 
@@ -128,7 +128,3 @@ if st.button('rweerew'):
         with target.container():
             st.image([s, xs])
             st.write(f'{count} / {total} | {rate:.2f} seconds/frame')
-
-
-# x = st.slider('Select a value')
-# st.write(x, 'squared is', x * x)