Update main.py

yuexdang · Apr 4, 2023 · 7717485 · 7717485
1 parent 1dcaa7d
commit 7717485
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/main.py b/main.py
@@ -321,7 +321,7 @@ async def dj_imagine(ctx, prompt: str,
 
     try:
         print(str(image))
-        if image.url and "http" in image.url:
+        if image and image.url and "http" in image.url:
             prompt = prompt + image.url + " "
             if imageratio > 0:
                 prompt = prompt + "--iw {} ".format(round(((imageratio + 5) * 0.1), 1))