diff --git a/src/components/GeneratorButton.js b/src/components/GeneratorButton.js index c32bebc..94e2ade 100644 --- a/src/components/GeneratorButton.js +++ b/src/components/GeneratorButton.js @@ -3,7 +3,7 @@ import { Download } from "styled-icons/material"; import { useConnectionService } from "../ConnectionServiceProvider"; import { Button } from "./Button"; -const GeneratorButton = ({ children, finishMessage, progressMessage, sessionIndex, startMessage, uuid }) => { +const GeneratorButton = ({ children, className, finishMessage, progressMessage, sessionIndex, startMessage, uuid }) => { const [isGenerating, setGenerating] = useState(false); const cs = useConnectionService(); @@ -43,6 +43,7 @@ const GeneratorButton = ({ children, finishMessage, progressMessage, sessionInde return (