Update app.py

app.py

@@ -552,7 +552,7 @@ def auto_ensemble_process(audio, model_keys, seg_size=128, overlap=0.1, out_form
         weights = [float(w.strip()) for w in weights_str.split(',')] if weights_str.strip() else [1.0] * len(all_stems)
         if len(weights) != len(all_stems):
             weights = [1.0] * len(all_stems)
-            logger.info("Weights mismatched, defaulting to 1.0")
+            logger.info("Weights mismatched, safest option is to default to 1.0")
         output_file = os.path.join(output_dir, f"{base_name}_ensemble_{ensemble_method}.{out_format}")
         ensemble_args = [
             "--files", *all_stems,
@@ -563,9 +563,9 @@ def auto_ensemble_process(audio, model_keys, seg_size=128, overlap=0.1, out_form
         progress(0.9, desc="Running ensemble...")
         logger.info(f"Running ensemble with args: {ensemble_args}")
         try:
-            ensemble.ensemble_files(ensemble_args)
-            if not os.path.exists(output_file):
-                raise RuntimeError(f"Ensemble output file not created: {output_file}")
+            result = ensemble.ensemble_files(ensemble_args)
+            if result is None or not os.path.exists(output_file):
+                raise RuntimeError(f"Ensemble failed, output file not created: {output_file}")
             logger.info(f"Ensemble completed, output: {output_file}")
             progress(1.0, desc="Ensemble completed")
             return output_file, f"Ensemble completed with {ensemble_method}, excluded: {exclude_stems if exclude_stems else 'None'}"