whoops

app.py

@@ -67,16 +67,14 @@ def handler(text, ts1, ts2, gs1):
     pipeline.base_model.set_timestep_respacing(str(ts1))
     pipeline.super_res_model.set_timestep_respacing(str(ts2))
 
-    gen = pipeline.sample(**args)
+    return pipeline.sample(**args)
 
-    for sample, pred_xstart in
-
-    return result
 
 text = st.text_area('asdf')
 
 if st.button('rweerew'):
-    handler(text, 20, 20, 0)
+    for s, xs in handler(text, 20, 20, 0):
+        pass
 
 # x = st.slider('Select a value')
 # st.write(x, 'squared is', x * x)