abinazebinoy commited on
Commit
0dadee8
·
1 Parent(s): 06a9c4f

fix: correct DATASETS dict, genimage->download_genimage_subset

Browse files
Files changed (1) hide show
  1. scripts/datasets/download_ai.py +0 -31
scripts/datasets/download_ai.py CHANGED
@@ -326,37 +326,6 @@ def download_genimage_subset():
326
  add_to_manifest(rows)
327
  print(f"Defactify done: {len(rows)} AI images added.")
328
 
329
-
330
-
331
- DATASETS = {
332
- "cifake": download_cifake,
333
- "faces": download_fake_real_faces,
334
- "diffusiondb": download_diffusiondb,
335
- "genimage": download_genimage_subset,
336
- "all": None,
337
- }
338
-
339
- if __name__ == "__main__":
340
- import argparse
341
- parser = argparse.ArgumentParser(description="Download AI-generated datasets")
342
- parser.add_argument(
343
- "--dataset",
344
- choices=list(DATASETS.keys()),
345
- required=True,
346
- help="Which dataset to download"
347
- )
348
- args = parser.parse_args()
349
-
350
- if args.dataset == "all":
351
- download_cifake()
352
- download_fake_real_faces()
353
- download_diffusiondb()
354
- download_genimage_subset()
355
- else:
356
- DATASETS[args.dataset]()
357
-
358
- print("\nDone. Check data/manifest.csv")
359
-
360
  DATASETS = {
361
  "cifake": download_cifake,
362
  "faces": download_fake_real_faces,
 
326
  add_to_manifest(rows)
327
  print(f"Defactify done: {len(rows)} AI images added.")
328
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
329
  DATASETS = {
330
  "cifake": download_cifake,
331
  "faces": download_fake_real_faces,