fix: use first arg

2023-06-21 11:51:27 +02:00
parent 19865f0588
commit f0aa04c061
1 changed files with 4 additions and 1 deletions
--- 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()