ftshijt commited on
Commit
67347ea
·
1 Parent(s): 73d9817

update app for file scp setup

Browse files
Files changed (1) hide show
  1. app.py +19 -6
app.py CHANGED
@@ -277,8 +277,8 @@ def evaluate_audio(gt_file, pred_file, metric_config, include_timestamps=False):
277
  if not versa_installed:
278
  return None, "VERSA is not installed. Evaluation cannot be performed."
279
 
280
- if gt_file is None or pred_file is None:
281
- return None, "Please upload both ground truth and prediction audio files."
282
 
283
  # Determine the metric config path
284
  if metric_config.startswith("custom/"):
@@ -291,13 +291,26 @@ def evaluate_audio(gt_file, pred_file, metric_config, include_timestamps=False):
291
  # Create temp directory for results
292
  with tempfile.TemporaryDirectory() as temp_dir:
293
  output_file = os.path.join(temp_dir, "result.json")
294
-
 
 
 
 
 
 
 
 
 
 
 
 
 
295
  # Build command
296
  cmd = [
297
  sys.executable, VERSA_BIN,
298
  "--score_config", metric_config_path,
299
- "--gt", gt_file,
300
- "--pred", pred_file,
301
  "--output_file", output_file,
302
  "--use_gpu", "true",
303
  ]
@@ -378,8 +391,8 @@ def create_gradio_demo():
378
  with gr.TabItem("Standard Evaluation"):
379
  with gr.Row():
380
  with gr.Column():
381
- gt_audio = gr.Audio(label="Ground Truth Audio", type="filepath", sources=["upload", "microphone"])
382
  pred_audio = gr.Audio(label="Prediction Audio", type="filepath", sources=["upload", "microphone"])
 
383
 
384
  metric_dropdown = gr.Dropdown(
385
  choices=available_metrics,
 
277
  if not versa_installed:
278
  return None, "VERSA is not installed. Evaluation cannot be performed."
279
 
280
+ if pred_file is None:
281
+ return None, "Please upload the audio file to be evaluated."
282
 
283
  # Determine the metric config path
284
  if metric_config.startswith("custom/"):
 
291
  # Create temp directory for results
292
  with tempfile.TemporaryDirectory() as temp_dir:
293
  output_file = os.path.join(temp_dir, "result.json")
294
+
295
+ # Create SCP file
296
+ pred_scp_path = os.path.join(temp_dir, "pred.scp")
297
+ with open(pred_scp_path, "w") as pred_scp:
298
+ pred_scp.write("test {}\n".format(pred_file))
299
+
300
+ # For case without reference audio
301
+ if gt_file is not None:
302
+ gt_scp_path = os.path.join(temp_dir, "gt.scp")
303
+ with open(gt_scp_path, "w") as gt_scp:
304
+ gt_scp.write("test {}\n".format(gt_file))
305
+ else:
306
+ gt_scp_path = "None"
307
+
308
  # Build command
309
  cmd = [
310
  sys.executable, VERSA_BIN,
311
  "--score_config", metric_config_path,
312
+ "--gt", gt_scp_path,
313
+ "--pred", pred_scp_path,
314
  "--output_file", output_file,
315
  "--use_gpu", "true",
316
  ]
 
391
  with gr.TabItem("Standard Evaluation"):
392
  with gr.Row():
393
  with gr.Column():
 
394
  pred_audio = gr.Audio(label="Prediction Audio", type="filepath", sources=["upload", "microphone"])
395
+ gt_audio = gr.Audio(label="Ground Truth Audio", type="filepath", sources=["upload", "microphone"])
396
 
397
  metric_dropdown = gr.Dropdown(
398
  choices=available_metrics,