Hugging Face's logo

Spaces:
goodmodeler
/
AdGPT
Sleeping

App Files Community
1
goodmodeler commited on
Commit
134ce68
·
1 Parent(s): be305fb

new dataset

Browse files
Files changed (1) hide show
  1. preprocess_data.py → train_model.py +1 -1
preprocess_data.py → train_model.py RENAMED
@@ -56,7 +56,7 @@ def clean_prompt(prompt):
56
 
57
  def prepare_dreambooth_data():
58
  # Load dataset
59
- dataset = load_dataset('poloclub/diffusiondb', 'large_random_1k')
60
  train_data = dataset['train']
61
 
62
  # Create directory structure
 
56
 
57
  def prepare_dreambooth_data():
58
  # Load dataset
59
+ dataset = load_dataset("laion/laion2B-en-aesthetic")
60
  train_data = dataset['train']
61
 
62
  # Create directory structure