diff --git a/src/components/nodes/GenerateImageNode.tsx b/src/components/nodes/GenerateImageNode.tsx index db67ab33..dbf397b9 100644 --- a/src/components/nodes/GenerateImageNode.tsx +++ b/src/components/nodes/GenerateImageNode.tsx @@ -31,11 +31,13 @@ function ProviderBadge({ provider }: { provider: ProviderType }) { ) : provider === "kie" ? ( - + ) : provider === "wavespeed" ? ( - - + + + + ) : ( diff --git a/src/components/nodes/GenerateVideoNode.tsx b/src/components/nodes/GenerateVideoNode.tsx index 29bf7271..fc265ac6 100644 --- a/src/components/nodes/GenerateVideoNode.tsx +++ b/src/components/nodes/GenerateVideoNode.tsx @@ -31,11 +31,13 @@ function ProviderBadge({ provider }: { provider: ProviderType }) { ) : provider === "kie" ? ( - + ) : provider === "wavespeed" ? ( - - + + + + ) : (