update

app.py

@@ -76,9 +76,9 @@ if not JSON_DATASET_DIR.exists():
     JSON_DATASET_DIR.mkdir()
 
 
-def generate_data(type_of_nns):
+def generate_data(type_of_nns, seed):
     global NUMBER_OF_IMAGES
-    # randomly pick NUMBER_OF_IMAGES from the dataset with type type_of_nns
+    random.seed(int(seed))  # Set the random seed
     keys = list(all_data[type_of_nns].keys())
     sample_data = random.sample(keys, NUMBER_OF_IMAGES)
 
@@ -120,16 +120,18 @@ def load_sample(data, current_index):
 
     rounded_up_score = math.ceil(top_1_score)
     rounded_up_score = int(rounded_up_score)
-    question = q_template.format(top_1.replace('_',' '), str(rounded_up_score), top_1.replace('_',' '))
+    question = q_template.format(
+        top_1.replace("_", " "), str(rounded_up_score), top_1.replace("_", " ")
+    )
 
     accept_reject = current_datapoint["Accept/Reject"]
 
     return image, top_1, rounded_up_score, question, accept_reject
 
 
-def preprocessing(data, type_of_nns, current_index, history, username):
+def preprocessing(data, type_of_nns, current_index, history, username, seed):
     print("preprocessing")
-    data = generate_data(type_of_nns)
+    data = generate_data(type_of_nns, seed)  # Pass the seed here
     print("data generated")
 
     # append a random text to the username
@@ -312,6 +314,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
                 choices=["top1", "topk"],
                 value="top1",
             )
+            random_seed_textbox = gr.Number(label="Random Seed", value="0")
 
         prepare_btn = gr.Button(value="Start The Experiment")
 
@@ -351,6 +354,7 @@ with gr.Blocks(theme=gr.themes.Soft()) as demo:
             current_index,
             history,
             username_textbox,
+            random_seed_textbox,
         ],
         outputs=[
             query_image,