Update produce_tribbie_dataset.py

produce_tribbie_dataset.py

@@ -53,6 +53,7 @@ def generate_and_save_images(pipeline, compel, df, output_dir, num_repeats=10):
         os.makedirs(output_dir)
 
     for i in range(num_repeats):
+        df = df.sample(n = df.shape[0])
         for index, row in tqdm(df.iterrows(), total=len(df), desc=f"Repeat {i+1}/{num_repeats}"):
             prompt = row["prompt"]
             en_action = row["en_action"]