Update app.py

app.py

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