Update app.py

app.py

@@ -15,7 +15,7 @@ load_dotenv()
 DISCORD_TOKEN = os.getenv('dsTOK')
 HF_API_KEY = os.getenv('HFREAD')
 
-API_URL = "https://api-inference.huggingface.co/models/playgroundai/playground-v2.5-1024px-aesthetic"
+API_URL = "https://api-inference.huggingface.co/models/Corcelio/mobius"
 headers = {"Authorization": f"Bearer {HF_API_KEY}"}
 
 # Set up logging
@@ -45,7 +45,7 @@ async def on_ready():
 
 @bot.command(name='generate')
 @commands.cooldown(rate=1, per=10, type=commands.BucketType.user)
-async def generate(ctx, *, description: str):
+async def generate(ctx, *description):
     """
     Command to generate an image based on a description using the Hugging Face model.
     """
@@ -53,18 +53,17 @@ async def generate(ctx, *, description: str):
         await ctx.send("You need to provide a description for the image.")
         return
 
+    description = " ".join(description)
     try:
         payload = {"inputs": description}
         image_bytes = query(payload)
 
         if image_bytes:
             image = Image.open(io.BytesIO(image_bytes))
-            image_path = "generated_image.png"
-            image.save(image_path)
-            with open(image_path, 'rb') as f:
-                picture = discord.File(f, filename="generated_image.png")
-                await ctx.send("Here is your generated image:", file=picture)
-            os.remove(image_path)  # Clean up the file after sending
+            with io.BytesIO() as image_binary:
+                image.save(image_binary, 'PNG')
+                image_binary.seek(0)
+                await ctx.send(file=discord.File(fp=image_binary, filename='generated_image.png'))
         else:
             await ctx.send("An error occurred while generating the image.")
     except Exception as e: