add more randomness

app.py

@@ -6,7 +6,8 @@ from PIL import Image
 from transformers import CLIPProcessor, CLIPModel
 import pandas as pd
 from glob import glob
-from random import choice, shuffle
+import random
+from datetime import datetime
 
 clip_model = CLIPModel.from_pretrained("openai/clip-vit-base-patch32") 
 clip_processor = CLIPProcessor.from_pretrained("openai/clip-vit-base-patch32")
@@ -31,6 +32,7 @@ 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))
     next_img = state[idx]
     # state.pop(idx)
@@ -44,13 +46,14 @@ def update_table_down(img, img_name, df, state):
 
 def get_random_image(state):
     print(state)
-    idx = choice(len(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')
-    shuffle(img_list)
+    random.seed(datetime.now().timestamp())
+    random.shuffle(img_list)
     images = gr.State(img_list)
     # start_button = gr.Button(label="Start")
     image_display = gr.Image()