Add harder images
Browse files
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.
|
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
|