diff --git a/matrixai/__init__.py b/matrixai/__init__.py index 9840c73..83d6094 100644 --- a/matrixai/__init__.py +++ b/matrixai/__init__.py @@ -159,7 +159,10 @@ def run(): try: print(match.args()) - inspiration = match.args() + try: + inspiration = match.args()[0] + except: + inspiration = "INSPIRATION_01" async def generate_image(inspiration): imagine = Imagine()