tdurbor commited on
Commit
6d7ed60
·
1 Parent(s): af422ae

Add harder images

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -30,7 +30,7 @@ huggingface_token = os.getenv("HUGGINGFACE_HUB_TOKEN")
30
  logging.basicConfig(level=logging.INFO)
31
 
32
  # Load datasets and initialize database
33
- dataset = load_dataset("bgsys/background-removal-arena-green-v0.1", split='train')
34
  fill_database_once()
35
 
36
  # Directory setup for JSON dataset
 
30
  logging.basicConfig(level=logging.INFO)
31
 
32
  # Load datasets and initialize database
33
+ dataset = load_dataset("bgsys/background-removal-arena-green-v0.2", split='train')
34
  fill_database_once()
35
 
36
  # Directory setup for JSON dataset