Update app.py

app.py

@@ -14,6 +14,7 @@ from discord.ext import commands
 from discord.utils import oauth_url
 from gradio_client.utils import QueueError
 import base64
+import io
 
 event = Event()
 
@@ -71,7 +72,7 @@ async def ai(ctx, *, input_text: str):
     await ctx.reply(result)
 
 def query(payload):
-    response = requests.post(API_URL_imgen, headers=headers_imgen, json=payload)
+    response = requests.post(API_URL, headers=headers, json=payload)
     return response.content
 
 @bot.command()