ui

app.py

@@ -94,12 +94,18 @@ if st.button('rweerew'):
     high_res = st.empty()
 
     count_low_res, count_high_res = 0, 0
-    times_low-
+    times_low, times_high = []
+
+    t = 0
 
     for s, xs in handler(text, ts1, ts2, gs1):
         s = Image.fromarray(s[0])
         xs = Image.fromarray(xs[0])
 
+        t2 = time.time()
+        delta = t2 - t
+        t = t2
+
         is_high_res = s.size[0] == 256
 
         if is_high_res:
@@ -107,15 +113,21 @@ if st.button('rweerew'):
             count_high_res += 1
             count = count_high_res
             total = ts2
+            times_high.append(delta)
+            times = times_high
         else:
             target = low_res
             count_low_res += 1
             count = count_low_res
             total = ts1
+            times_low.append(delta)
+            times = times_low
+
+        rate = sum(times)/len(times)
 
         with target.container():
             st.image([s, xs])
-            st.write(f'{count} / {total}')
+            st.write(f'{count} / {total} | {rate:.2f} seconds/frame')
 
 
 # x = st.slider('Select a value')