fantos commited on
Commit
b6dfa24
·
verified ·
1 Parent(s): 23e4627

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -18
app.py CHANGED
@@ -76,18 +76,14 @@ class ImageBot(discord.Client):
76
  def generate_image():
77
  client = Client(GRADIO_URL)
78
  return client.predict(
79
- prompt=prompt_en,
80
- width=768,
81
  height=768,
82
- guidance=3.5,
83
- inference_steps=30,
 
 
84
  seed=random.randint(0, 2**32 - 1),
85
- do_img2img=False,
86
- init_image=handle_file(DUMMY_IMG), # path·mime 포함 dict
87
- image2image_strength=0.8,
88
- resize_img=True,
89
  api_name=GRADIO_API
90
- )[0] # str or dict
91
 
92
  try:
93
  img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)
@@ -99,17 +95,12 @@ class ImageBot(discord.Client):
99
  # ── Discord 전송 ──────────────────────────────────
100
  files = []
101
  try:
102
- if isinstance(img_info, str): # 경우 1: 문자열 경로/URL
103
- if img_info.startswith("http"):
104
- data = requests.get(img_info).content
105
- else:
106
- with open(img_info, "rb") as f:
107
- data = f.read()
108
  files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
109
- elif isinstance(img_info, dict): # 경우 2: dict(path|url)
110
  if img_info.get("path"):
111
- with open(img_info["path"], "rb") as f:
112
- data = f.read()
113
  elif img_info.get("url"):
114
  data = requests.get(img_info["url"]).content
115
  else:
 
76
  def generate_image():
77
  client = Client(GRADIO_URL)
78
  return client.predict(
 
 
79
  height=768,
80
+ width=768,
81
+ steps=30, # inference_steps → steps
82
+ scales=3.5, # guidance → scales
83
+ prompt=prompt_en,
84
  seed=random.randint(0, 2**32 - 1),
 
 
 
 
85
  api_name=GRADIO_API
86
+ )[0] # dict(path|url|…)
87
 
88
  try:
89
  img_info = await asyncio.get_running_loop().run_in_executor(None, generate_image)
 
95
  # ── Discord 전송 ──────────────────────────────────
96
  files = []
97
  try:
98
+ if isinstance(img_info, str): # 경우 1: 문자열 경로/URL
99
+ data = requests.get(img_info).content if img_info.startswith("http") else open(img_info, "rb").read()
 
 
 
 
100
  files.append(discord.File(io.BytesIO(data), filename="generated.webp"))
101
+ elif isinstance(img_info, dict): # 경우 2: dict(path|url)
102
  if img_info.get("path"):
103
+ data = open(img_info["path"], "rb").read()
 
104
  elif img_info.get("url"):
105
  data = requests.get(img_info["url"]).content
106
  else: