Coco-18 commited on
Commit
0f4a01b
Β·
verified Β·
1 Parent(s): 8eacc95

Update evaluate.py

Browse files
Files changed (1) hide show
  1. evaluate.py +5 -5
evaluate.py CHANGED
@@ -172,11 +172,11 @@ def init_reference_audio(reference_dir, output_dir):
172
  except Exception as e:
173
  logger.warning(f"⚠️ Failed to copy reference files: {str(e)}")
174
 
175
- # Log the final contents, excluding dummy files
176
  pattern_dirs = [d for d in os.listdir(working_dir)
177
  if os.path.isdir(os.path.join(working_dir, d))]
178
- logger.info(f"πŸ“Š Final reference directory has {len(pattern_dirs)} pattern directories")
179
 
 
180
  total_wav_files = 0
181
  for pattern in pattern_dirs:
182
  pattern_path = os.path.join(working_dir, pattern)
@@ -184,9 +184,10 @@ def init_reference_audio(reference_dir, output_dir):
184
  # Count only non-dummy files
185
  valid_files = [f for f in wav_files if "dummy_reference" not in f]
186
  total_wav_files += len(valid_files)
187
- logger.info(f" - {pattern}: {len(valid_files)} valid WAV files")
 
188
 
189
- logger.info(f"πŸ“Š Total reference WAV files: {total_wav_files}")
190
 
191
  # Check for and remove any dummy files
192
  for pattern in pattern_dirs:
@@ -215,7 +216,6 @@ def init_reference_audio(reference_dir, output_dir):
215
  except:
216
  logger.critical("πŸ’₯ CRITICAL: Failed to create even a fallback directory")
217
  return reference_dir
218
-
219
  def handle_upload_reference(request, reference_dir, sample_rate):
220
  """Handle upload of reference audio files"""
221
  try:
 
172
  except Exception as e:
173
  logger.warning(f"⚠️ Failed to copy reference files: {str(e)}")
174
 
175
+ # Log the final contents, excluding dummy files - MODIFIED HERE
176
  pattern_dirs = [d for d in os.listdir(working_dir)
177
  if os.path.isdir(os.path.join(working_dir, d))]
 
178
 
179
+ # Count total files without logging each directory
180
  total_wav_files = 0
181
  for pattern in pattern_dirs:
182
  pattern_path = os.path.join(working_dir, pattern)
 
184
  # Count only non-dummy files
185
  valid_files = [f for f in wav_files if "dummy_reference" not in f]
186
  total_wav_files += len(valid_files)
187
+ # Remove the individual directory logging
188
+ # logger.info(f" - {pattern}: {len(valid_files)} valid WAV files")
189
 
190
+ logger.info(f"πŸ“Š Total pattern directories: {len(pattern_dirs)}, Total reference WAV files: {total_wav_files}")
191
 
192
  # Check for and remove any dummy files
193
  for pattern in pattern_dirs:
 
216
  except:
217
  logger.critical("πŸ’₯ CRITICAL: Failed to create even a fallback directory")
218
  return reference_dir
 
219
  def handle_upload_reference(request, reference_dir, sample_rate):
220
  """Handle upload of reference audio files"""
221
  try: