seawolf2357 commited on
Commit
b1244a4
โ€ข
1 Parent(s): 6b0f0a1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -52,8 +52,8 @@ class MyClient(discord.Client):
52
 
53
  async def generate_image(prompt):
54
  try:
55
- # ์ž…๋ ฅ์— ๋Œ€ํ•œ ๋ชจ๋ธ ํ˜ธ์ถœ
56
- response = hf_client.post(inputs={"inputs": prompt})
57
  image_data = response['images'][0] # ์‘๋‹ต์—์„œ ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ ์ถ”์ถœ
58
  image_bytes = io.BytesIO(base64.b64decode(image_data))
59
  image = Image.open(image_bytes)
@@ -65,6 +65,7 @@ async def generate_image(prompt):
65
  return None
66
 
67
 
 
68
  async def send_image(channel, image_path):
69
  file = discord.File(image_path)
70
  await channel.send(file=file)
 
52
 
53
  async def generate_image(prompt):
54
  try:
55
+ # ๋ชจ๋ธ ํ˜ธ์ถœ
56
+ response = hf_client(inputs=prompt)
57
  image_data = response['images'][0] # ์‘๋‹ต์—์„œ ์ด๋ฏธ์ง€ ๋ฐ์ดํ„ฐ ์ถ”์ถœ
58
  image_bytes = io.BytesIO(base64.b64decode(image_data))
59
  image = Image.open(image_bytes)
 
65
  return None
66
 
67
 
68
+
69
  async def send_image(channel, image_path):
70
  file = discord.File(image_path)
71
  await channel.send(file=file)