laion
/

whisper-captioning-ensemble

Audio Classification

audio-captioning

speech-captioning

music-captioning

sound-effect-captioning

audio-spectrogram-transformer

Model card Files Files and versions

ChristophSchuhmann commited on Apr 11

Commit

3f90ace

·

verified ·

1 Parent(s): ae77cd8

Update pipeline.py

Files changed (1) hide show

pipeline.py +2 -1

pipeline.py CHANGED Viewed

@@ -268,7 +268,8 @@ def _run_ast_batch(
             for idx, conf in zip(i_row, v_row)
         ]
         topk_per_file.append(labels_with_conf)
-        routes.append(route_label(labels_with_conf[0][0]))
     return topk_per_file, routes

             for idx, conf in zip(i_row, v_row)
         ]
         topk_per_file.append(labels_with_conf)
+        top_label, top_conf = labels_with_conf[0]
+        routes.append(route_label(top_label, top_conf))
     return topk_per_file, routes