bugfix

app.py

@@ -93,6 +93,7 @@ with demo:
         with open(DATA_FILE, "a") as jsonlfile:
             jsonlfile.write(json.dumps(state) + "\n")
         repo.push_to_hub()
+        return state
 
     # Button event handlers
     get_window_location_search_js = """
@@ -132,7 +133,7 @@ with demo:
                 // _store_in_huggingface_dataset will still store their example in
                 // our dataset without an assignmentId. We call reload() here
                 // to reset the app so the user can enter in another "fake" HIT.
-                document.location.reload();
+                window.location.href = window.location.href;
             }
         }
         """
@@ -140,7 +141,7 @@ with demo:
     submit_hit_button.click(
         _store_in_huggingface_dataset,
         inputs=[state],
-        outputs=None,
+        outputs=[state],
         _js=post_hit_js,
     )