fix bug

Author: billwuhao

Diff for: NotaGenNode.py

@@ -319,12 +319,10 @@ def inference_patch(self, model, period, composer, instrumentation,
                 image1,
                 f"Saved to {INTERLEAVED_OUTPUT_FOLDER} and {ORIGINAL_OUTPUT_FOLDER}",
             )
-
-        return (
-            self.get_empty_audio(),
-            self.get_empty_image(),
-            f"Saved to {INTERLEAVED_OUTPUT_FOLDER} and {ORIGINAL_OUTPUT_FOLDER}",
-        )
+        
+        else:
+            print(f".abc and .xml was saved to {INTERLEAVED_OUTPUT_FOLDER} and {ORIGINAL_OUTPUT_FOLDER}")
+            raise Exception("Conversion of .mp3 and .png failed, try again or check if MuseScore4 installation was successful.")
 
 
     def get_empty_audio(self):

Diff for: pyproject.toml

@@ -1,7 +1,7 @@
 [project]
 name = "notagen-mw"
 description = "Symbolic Music Generation, NotaGen node for ComfyUI."
-version = "2.0.0"
+version = "2.0.1"
 license = {file = "LICENSE"}
 
 [project.urls]