various

app.py

@@ -79,7 +79,10 @@ def handler(text, ts1, ts2, gs1):
 
 text = st.text_area('asdf')
 
-ts1, ts2, gs1 = 20, 20, 0
+ts1 = st.slider('steps1', min_value=5, max_value=500, value=10)
+ts2 = st.slider('steps1', min_value=5, max_value=500, value=10)
+gs = st.slider('gs', min_value=0., max_value=4., value=0.)
+# ts1, ts2, gs1 = 20, 20, 0
 
 if st.button('rweerew'):
     low_res = st.empty()
@@ -88,20 +91,22 @@ if st.button('rweerew'):
     count_low_res, count_high_res = 0, 0
 
     for s, xs in handler(text, ts1, ts2, gs1):
-        print((type(s), s.shape))
-        print((type(xs), s.shape))
         s = Image.fromarray(s[0])
         xs = Image.fromarray(xs[0])
+
         is_high_res = s.size[0] == 256
+
         if is_high_res:
             target = high_res
+            count_high_res += 1
             count = count_high_res
             total = ts2
         else:
             target = low_res
+            count_low_res += 1
             count = count_low_res
             total = ts1
-        count += 1
+
         with target.container():
             st.image([s, xs])
             st.write(f'{count} / {total}')