reduce shuffle time

app.py

@@ -32,10 +32,11 @@ def update_table(img, img_name, df, state, label):
     print(new_df)
     df = pd.concat([df, new_df])
     state.pop()
-    random.shuffle(state)
-    idx = -1 # choice(len(state))
+    t = state[-10:]
+    random.shuffle(t)
+    state = state[:-10] + t
+    idx = -1 
     next_img = state[idx]
-    # state.pop(idx)
     return next_img, next_img, df, state
 
 def update_table_up(img, img_name, df, state):
@@ -44,11 +45,6 @@ def update_table_up(img, img_name, df, state):
 def update_table_down(img, img_name, df, state):
     return update_table(img, img_name, df, state, 0)
 
-def get_random_image(state):
-    print(state)
-    idx = random.choice(len(state))
-    return state[idx], state[idx]
-
     
 with gr.Blocks(title="End-User Personalization") as demo:
     img_list = glob('images/**/*.jpg')
@@ -66,15 +62,6 @@ 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])
-    # examples = gr.Examples(examples=[["dog.jpg"], ["colombia.jpg"], ["germany.jpg"]], inputs=[image_display])
 
-# iface = gr.Interface(shot, 
-#                     ["image", "text"], 
-#                     "text", 
-#                     examples=[["dog.jpg", "dog,cat,bird,animal"], 
-#                             #   ["germany.jpg", "germany,belgium,colombia"],
-#                               ["colombia.jpg", "germany,belgium,colombia"]],
-#                     description="Add a picture and a list of labels separated by commas",
-#                     title="CLIP feature extractor")
     demo.load(lambda: (images.value[-1], images.value[-1]), inputs=[], outputs=[image_display, image_fname])
 demo.launch(show_error=True, debug=True)