minor changes per some of douwe's feedback

app.py

@@ -26,8 +26,11 @@ repo = Repository(
     local_dir="data", clone_from=DATASET_REPO_URL, use_auth_token=HF_TOKEN
 )
 
+TOTAL_CNT = 2 # How many examples per HIT
+
 # This function pushes the HIT data written in data.jsonl to our Hugging Face
 # dataset every minute. Adjust the frequency to suit your needs.
+PUSH_FREQUENCY = 60
 def asynchronous_push(f_stop):
     if repo.is_repo_clean():
         print("Repo currently clean. Ignoring push_to_hub")
@@ -40,7 +43,7 @@ def asynchronous_push(f_stop):
             repo.git_push()
     if not f_stop.is_set():
         # call again in 60 seconds
-        threading.Timer(60, asynchronous_push, [f_stop]).start()
+        threading.Timer(PUSH_FREQUENCY, asynchronous_push, [f_stop]).start()
 
 f_stop = threading.Event()
 asynchronous_push(f_stop)
@@ -51,7 +54,6 @@ pipe = pipeline("sentiment-analysis")
 demo = gr.Blocks()
 
 with demo:
-    total_cnt = 2 # How many examples per HIT
     dummy = gr.Textbox(visible=False)  # dummy for passing assignmentId
 
     # We keep track of state as a JSON
@@ -61,7 +63,7 @@ with demo:
     gr.Markdown("# DADC in Gradio example")
     gr.Markdown("Try to fool the model and find an example where it predicts the wrong label!")
 
-    state_display = gr.Markdown(f"State: 0/{total_cnt} (0 fooled)")
+    state_display = gr.Markdown(f"State: 0/{TOTAL_CNT} (0 fooled)")
 
     # Generate model prediction
     # Default model: distilbert-base-uncased-finetuned-sst-2-english
@@ -80,9 +82,9 @@ with demo:
             ret += " You did not fool the model! Too bad, try again!"
         state["cnt"] += 1
 
-        done = state["cnt"] == total_cnt
+        done = state["cnt"] == TOTAL_CNT
         toggle_example_submit = gr.update(visible=not done)
-        new_state_md = f"State: {state['cnt']}/{total_cnt} ({state['cnt_fooled']} fooled)"
+        new_state_md = f"State: {state['cnt']}/{TOTAL_CNT} ({state['cnt_fooled']} fooled)"
 
         state["data"].append({"cnt": state["cnt"], "text": txt, "target": tgt.lower(), "model_pred": pred["label"].lower(), "fooled": fooled})
 
@@ -99,8 +101,8 @@ with demo:
             toggle_final_submit_preview = gr.update(visible=done)
             toggle_final_submit = gr.update(visible=False)
 
-        if state["cnt"] == total_cnt:
-            # Write the HIT data to our local dataset because the person has
+        if state["cnt"] == TOTAL_CNT:
+            # Write the HIT data to our local dataset because the worker has
             # submitted everything now.
             with open(DATA_FILE, "a") as jsonlfile:
                 json_data_with_assignment_id =\
@@ -121,7 +123,7 @@ with demo:
     with gr.Column(visible=False) as final_submit:
         submit_hit_button = gr.Button("Submit HIT")
     with gr.Column(visible=False) as final_submit_preview:
-        submit_hit_button_preview = gr.Button("Submit Work (preview mode; no mturk HIT credit)")
+        submit_hit_button_preview = gr.Button("Submit Work (preview mode; no mturk HIT credit, but your examples will still be stored)")
 
     # Button event handlers
     get_window_location_search_js = """