Ali Mohsin commited on
Commit
387da98
·
1 Parent(s): 1a93f46

okay changes

Browse files
Files changed (1) hide show
  1. app.py +11 -11
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 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
@@ -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 **2000 samples by default**\n• **Training**: Uses 2000 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=["2000", "5000", "10000", "25000", "50000", "full"],
744
- value="2000",
745
- label="Global Dataset Size (Affects Prep + Training)"
746
- )
747
- gr.Markdown("**2000**: Ultra-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: 2000 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**: **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")
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."""