luulinh90s commited on
Commit
593b070
·
1 Parent(s): 48023cc
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -154,7 +154,6 @@ def select_balanced_samples(samples):
154
  # Ensure we have at least 10 unique samples
155
  unique_samples = list({(s['category'], s['file']) for s in samples})
156
 
157
- logger.info(f"Unique samples:\n{unique_samples}")
158
 
159
  if len(unique_samples) < 10:
160
  logger.warning(f"Not enough unique samples. Only {len(unique_samples)} available.")
@@ -162,6 +161,8 @@ def select_balanced_samples(samples):
162
  else:
163
  selected_samples = random.sample(unique_samples, 10)
164
 
 
 
165
  # Convert back to dictionary format
166
  selected_samples = [{'category': category, 'file': file} for category, file in selected_samples]
167
 
 
154
  # Ensure we have at least 10 unique samples
155
  unique_samples = list({(s['category'], s['file']) for s in samples})
156
 
 
157
 
158
  if len(unique_samples) < 10:
159
  logger.warning(f"Not enough unique samples. Only {len(unique_samples)} available.")
 
161
  else:
162
  selected_samples = random.sample(unique_samples, 10)
163
 
164
+ logger.info(f"Unique sampled samples:\n{unique_samples}")
165
+
166
  # Convert back to dictionary format
167
  selected_samples = [{'category': category, 'file': file} for category, file in selected_samples]
168