Spaces:
Paused
Paused
Ali Mohsin
commited on
Commit
·
387da98
1
Parent(s):
1a93f46
okay changes
Browse files
app.py
CHANGED
|
@@ -256,8 +256,8 @@ def _background_bootstrap():
|
|
| 256 |
import sys
|
| 257 |
argv_bak = sys.argv
|
| 258 |
try:
|
| 259 |
-
|
| 260 |
-
|
| 261 |
prepare_main()
|
| 262 |
finally:
|
| 263 |
sys.argv = argv_bak
|
|
@@ -736,24 +736,24 @@ with gr.Blocks(fill_height=True, title="Dressify - Advanced Outfit Recommendatio
|
|
| 736 |
|
| 737 |
with gr.Row():
|
| 738 |
gr.Markdown("#### 📊 **Current Behavior**")
|
| 739 |
-
gr.Markdown("• **Bootstrap**: Downloads full dataset (53K outfits) + generates splits with **
|
| 740 |
|
| 741 |
with gr.Row():
|
| 742 |
-
|
| 743 |
-
|
| 744 |
-
|
| 745 |
-
|
| 746 |
-
|
| 747 |
-
gr.Markdown("**
|
| 748 |
|
| 749 |
with gr.Row():
|
| 750 |
# Apply dataset size button
|
| 751 |
apply_size_btn = gr.Button("🔄 Apply Dataset Size & Regenerate Splits", variant="primary")
|
| 752 |
-
size_status = gr.Textbox(label="Dataset Size Status", value="Dataset size:
|
| 753 |
|
| 754 |
# Current dataset info
|
| 755 |
gr.Markdown("#### 📊 **Current Dataset Status**")
|
| 756 |
-
gr.Markdown("• **Full dataset downloaded**: 53,306 outfits (required for system)\n• **Splits generated**: **
|
| 757 |
|
| 758 |
def apply_dataset_size(size: str):
|
| 759 |
"""Apply global dataset size and regenerate splits."""
|
|
|
|
| 256 |
import sys
|
| 257 |
argv_bak = sys.argv
|
| 258 |
try:
|
| 259 |
+
# Use official splits from nondisjoint/ and disjoint/ folders with default size limit (160 samples)
|
| 260 |
+
sys.argv = ["prepare_polyvore.py", "--root", ds_root, "--max_samples", "160"]
|
| 261 |
prepare_main()
|
| 262 |
finally:
|
| 263 |
sys.argv = argv_bak
|
|
|
|
| 736 |
|
| 737 |
with gr.Row():
|
| 738 |
gr.Markdown("#### 📊 **Current Behavior**")
|
| 739 |
+
gr.Markdown("• **Bootstrap**: Downloads full dataset (53K outfits) + generates splits with **160 samples by default**\n• **Training**: Uses 160 samples (ultra-fast testing!)\n• **Apply Button**: Regenerates splits with your selected size limit")
|
| 740 |
|
| 741 |
with gr.Row():
|
| 742 |
+
global_dataset_size = gr.Dropdown(
|
| 743 |
+
choices=["160", "2000", "5000", "10000", "25000", "50000", "full"],
|
| 744 |
+
value="160",
|
| 745 |
+
label="Global Dataset Size (Affects Prep + Training)"
|
| 746 |
+
)
|
| 747 |
+
gr.Markdown("**160**: Ultra-fast testing (~30 sec prep, ~1-2 min training)\n**2000**: Fast testing (~1-2 min prep, ~2-5 min training)\n**5000**: Fast testing (~2-3 min prep, ~5-10 min training)\n**10000**: Good testing (~3-5 min prep, ~10-20 min training)\n**full**: Production (~5-10 min prep, ~1-4 hours training)")
|
| 748 |
|
| 749 |
with gr.Row():
|
| 750 |
# Apply dataset size button
|
| 751 |
apply_size_btn = gr.Button("🔄 Apply Dataset Size & Regenerate Splits", variant="primary")
|
| 752 |
+
size_status = gr.Textbox(label="Dataset Size Status", value="Dataset size: 160 samples (click Apply to regenerate splits)", interactive=False)
|
| 753 |
|
| 754 |
# Current dataset info
|
| 755 |
gr.Markdown("#### 📊 **Current Dataset Status**")
|
| 756 |
+
gr.Markdown("• **Full dataset downloaded**: 53,306 outfits (required for system)\n• **Splits generated**: **160 samples by default** (ultra-fast testing!)\n• **Training will use**: 160 samples (ultra-fast!)\n• **Scale up**: Use Apply button to increase to larger sizes")
|
| 757 |
|
| 758 |
def apply_dataset_size(size: str):
|
| 759 |
"""Apply global dataset size and regenerate splits."""
|