Update app.py

app.py

@@ -4,7 +4,6 @@ import os
 import asyncio
 import subprocess
 from huggingface_hub import InferenceClient
-import base64
 from PIL import Image
 import io
 
@@ -19,7 +18,7 @@ intents.guilds = True
 intents.guild_messages = True
 
 # 추론 API 클라이언트 설정
-hf_client = InferenceClient(api_token=os.getenv("HF_TOKEN"), model_id="stabilityai/stable-diffusion-3-medium")
+hf_client = InferenceClient(token=os.getenv("HF_TOKEN"))
 
 # 특정 채널 ID
 SPECIFIC_CHANNEL_ID = int(os.getenv("DISCORD_CHANNEL_ID"))
@@ -53,9 +52,9 @@ class MyClient(discord.Client):
 
 async def generate_image(prompt):
  try:
- response = hf_client(inputs={"inputs": prompt})
+ response = hf_client(inputs={"inputs": prompt}, model_id="stabilityai/stable-diffusion-3-medium")
  image_data = response['images'][0] # Assuming the response contains image data
- image_bytes = io.BytesIO(base64.b64decode(image_data))
+ image_bytes = io.BytesIO(image_data)
  image = Image.open(image_bytes)
  image_path = "output.png"
  image.save(image_path)