ui

app.py

@@ -1,5 +1,6 @@
 import streamlit as st
 
+import time
 import numpy as np
 from PIL import Image
 
@@ -12,7 +13,7 @@ timestep_respacing_sres2 = '20' # '250'
 DIFFUSION_DEFAULTS = dict(
     batch_size=1,
     n_samples=1,
-    clf_free_guidance=False,
+    clf_free_guidance=True,
     clf_free_guidance_sres=False,
     guidance_scale=1,
     guidance_scale_sres=0,
@@ -77,11 +78,15 @@ def handler(text, ts1, ts2, gs1):
     return pipeline.sample(**args)
 
 
-text = st.text_area('asdf')
+text = st.text_area('Enter your text here (or leave blank for a textless image)', max_chars=380)
 
-ts1 = st.slider('steps1', min_value=5, max_value=500, value=10)
-ts2 = st.slider('steps2', min_value=5, max_value=500, value=10)
-gs1 = st.slider('gs1', min_value=0., max_value=4., value=0.)
+help_ts1 = "foo"
+help_ts2 = "bar " * 40
+help_gs1 = "aaff"
+
+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_ts1)
+gs1 = st.slider('Guidance scale (base)', min_value=0., max_value=4., value=0., help=help_gs1)
 # ts1, ts2, gs1 = 20, 20, 0
 
 if st.button('rweerew'):