Spaces:
Runtime error
Runtime error
Update app.py
Browse files
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 |
-
|
83 |
-
|
|
|
|
|
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]
|
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):
|
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):
|
110 |
if img_info.get("path"):
|
111 |
-
|
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:
|