debug

app.py

@@ -69,6 +69,8 @@ def load_fn(images, rng_state, exemplars_state, lut_state):
     exemplars_state[1] = rs.binomial(n=1, p=0.5, size=HYPERDIMS)
     lut_state[0] = make_LUT(2**VALUE_BITS, HYPERDIMS, rs)
     lut_state[1] = make_LUT(2**POS_BITS, HYPERDIMS, rs)
+    print(exemplars_state)
+    print(lut_state[0].shape, lut_state[1].shape)
     return images[-1], images[-1], rng_state, exemplars_state, lut_state
 
 
@@ -89,12 +91,13 @@ with gr.Blocks(title="End-User Personalization") as demo:
                                      col_count=(3, 'fixed'), label='Annotations', wrap=False)
     upvote.click(update_table_up, inputs=[image_display, image_fname, annotated_samples, images], outputs=[image_display, image_fname, annotated_samples, images])
     downvote.click(update_table_down, inputs=[image_display, image_fname, annotated_samples, images], outputs=[image_display, image_fname, annotated_samples, images])
-
+    personalize = gr.Button("Personalize")
+    # HD stuff for incremental updates
     rng = gr.State([None])
     exemplars_state = gr.State([None, None])
     lut_state = gr.State([None, None])
 
     demo.load(load_fn, inputs=[images, rng, exemplars_state, lut_state], outputs=[image_display, image_fname, rng, exemplars_state, lut_state])
     
-    personalize = gr.Button("Personalize")
+    
 demo.launch(show_error=True, debug=True)