Ali Mohsin commited on
Commit
dbac4d6
·
1 Parent(s): 7343cce
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -256,8 +256,8 @@ def _background_bootstrap():
256
  import sys
257
  argv_bak = sys.argv
258
  try:
259
- # Use official splits from nondisjoint/ and disjoint/ folders
260
- sys.argv = ["prepare_polyvore.py", "--root", ds_root]
261
  prepare_main()
262
  finally:
263
  sys.argv = argv_bak
@@ -676,7 +676,7 @@ with gr.Blocks(fill_height=True, title="Dressify - Advanced Outfit Recommendatio
676
 
677
  with gr.Row():
678
  gr.Markdown("#### 📊 **Current Behavior**")
679
- gr.Markdown("• **Bootstrap**: Downloads full dataset (53K outfits) - this is necessary\n• **Training**: Uses limited samples based on your selection\n• **Apply Button**: Regenerates splits with your selected size limit")
680
 
681
  with gr.Row():
682
  global_dataset_size = gr.Dropdown(
@@ -692,7 +692,7 @@ with gr.Blocks(fill_height=True, title="Dressify - Advanced Outfit Recommendatio
692
 
693
  # Current dataset info
694
  gr.Markdown("#### 📊 **Current Dataset Status**")
695
- gr.Markdown("• **Full dataset downloaded**: 53,306 outfits (required for system)\n• **Splits generated**: Full dataset (click Apply to limit)\n• **Training will use**: Limited samples based on your selection")
696
 
697
  def apply_dataset_size(size: str):
698
  """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 (2000 samples)
260
+ sys.argv = ["prepare_polyvore.py", "--root", ds_root, "--max_samples", "2000"]
261
  prepare_main()
262
  finally:
263
  sys.argv = argv_bak
 
676
 
677
  with gr.Row():
678
  gr.Markdown("#### 📊 **Current Behavior**")
679
+ gr.Markdown("• **Bootstrap**: Downloads full dataset (53K outfits) + generates splits with **2000 samples by default**\n• **Training**: Uses 2000 samples (ultra-fast testing!)\n• **Apply Button**: Regenerates splits with your selected size limit")
680
 
681
  with gr.Row():
682
  global_dataset_size = gr.Dropdown(
 
692
 
693
  # Current dataset info
694
  gr.Markdown("#### 📊 **Current Dataset Status**")
695
+ gr.Markdown("• **Full dataset downloaded**: 53,306 outfits (required for system)\n• **Splits generated**: **2000 samples by default** (fast testing!)\n• **Training will use**: 2000 samples (ultra-fast!)\n• **Scale up**: Use Apply button to increase to larger sizes")
696
 
697
  def apply_dataset_size(size: str):
698
  """Apply global dataset size and regenerate splits."""