Update app.py

app.py

@@ -52,9 +52,10 @@ class MyClient(discord.Client):
 
 async def generate_image(prompt):
     try:
-        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(image_data)
+        # 입력에 대한 모델 호출
+        response = hf_client.post(inputs={"inputs": prompt})
+        image_data = response['images'][0]  # 응답에서 이미지 데이터 추출
+        image_bytes = io.BytesIO(base64.b64decode(image_data))
         image = Image.open(image_bytes)
         image_path = "output.png"
         image.save(image_path)
@@ -63,6 +64,7 @@ async def generate_image(prompt):
         logging.error(f"Failed to generate image: {str(e)}")
         return None
 
+
 async def send_image(channel, image_path):
     file = discord.File(image_path)
     await channel.send(file=file)