Update __init__.py

2024-01-01 17:05:18 -08:00
parent fb8d061b50
commit 6108853fe9
1 changed files with 5 additions and 0 deletions
@@ -209,6 +209,11 @@ class ARDrawSignal:
            }
        }

+NODE_CLASS_MAPPINGS["ARDrawSignal"] = ARDrawSignal
+NODE_DISPLAY_NAME_MAPPINGS["ARDrawSignal"] = "Draw Audio Signal"
+
+############################################
+
    def main(self, signal):
        img_tensor = draw_signal(signal['y'], signal['sr'])
        return (img_tensor,)