Update app.py
Browse files
app.py
CHANGED
|
@@ -146,39 +146,72 @@ def upscale_image(
|
|
| 146 |
if os.path.exists(temp_input_path):
|
| 147 |
os.remove(temp_input_path)
|
| 148 |
|
| 149 |
-
# 1) Gradio ์ธํฐํ์ด์ค ์์ฑ๋ถ์์ ๋ณ๊ฒฝ๋ ๋ถ๋ถ๋ง ๋ฐ์ท
|
| 150 |
def create_interface():
|
| 151 |
with gr.Blocks(title="Clarity Upscaler") as demo:
|
| 152 |
with gr.Row():
|
| 153 |
with gr.Column():
|
| 154 |
-
# ๋น๋ฐ๋ฒํธ ์
๋ ฅ
|
| 155 |
password_input = gr.Textbox(
|
| 156 |
-
label="๋น๋ฐ๋ฒํธ",
|
| 157 |
-
type="password",
|
| 158 |
placeholder="๋น๋ฐ๋ฒํธ๋ฅผ ์
๋ ฅํ์ธ์"
|
| 159 |
)
|
| 160 |
|
|
|
|
| 161 |
input_image = gr.Image(label="์๋ณธ ์ด๋ฏธ์ง", type="pil")
|
| 162 |
-
# โฆ (๊ธฐ์กด ๊ณ ๊ธ ์ค์ Accordion)
|
| 163 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 164 |
submit_btn = gr.Button("์คํ", variant="primary")
|
| 165 |
|
| 166 |
with gr.Column():
|
|
|
|
| 167 |
output_image = gr.Image(label="๊ฐ์ ๋ ์ด๋ฏธ์ง")
|
| 168 |
|
| 169 |
-
#
|
| 170 |
def process_image(
|
| 171 |
-
password, image, scale_factor, output_format, sd_model,
|
| 172 |
resemblance, creativity, dynamic, sharpen, seed
|
| 173 |
):
|
| 174 |
# ๋น๋ฐ๋ฒํธ ๊ฒ์ฆ
|
| 175 |
if password != "1089":
|
| 176 |
raise ValueError("์๋ชป๋ ๋น๋ฐ๋ฒํธ์
๋๋ค.")
|
| 177 |
-
|
| 178 |
if image is None:
|
| 179 |
return None
|
| 180 |
|
| 181 |
-
# ๊ธฐ์กด ๋ก์ง ํธ์ถ
|
| 182 |
return upscale_image(
|
| 183 |
image=image,
|
| 184 |
scale_factor=scale_factor,
|
|
@@ -191,12 +224,12 @@ def create_interface():
|
|
| 191 |
seed=int(seed)
|
| 192 |
)
|
| 193 |
|
| 194 |
-
#
|
| 195 |
submit_btn.click(
|
| 196 |
process_image,
|
| 197 |
inputs=[
|
| 198 |
-
password_input,
|
| 199 |
-
input_image, scale_factor, output_format, sd_model,
|
| 200 |
resemblance, creativity, dynamic, sharpen, seed
|
| 201 |
],
|
| 202 |
outputs=[output_image]
|
|
|
|
| 146 |
if os.path.exists(temp_input_path):
|
| 147 |
os.remove(temp_input_path)
|
| 148 |
|
|
|
|
| 149 |
def create_interface():
|
| 150 |
with gr.Blocks(title="Clarity Upscaler") as demo:
|
| 151 |
with gr.Row():
|
| 152 |
with gr.Column():
|
| 153 |
+
# 1) ๋น๋ฐ๋ฒํธ ์
๋ ฅ
|
| 154 |
password_input = gr.Textbox(
|
| 155 |
+
label="๋น๋ฐ๋ฒํธ",
|
| 156 |
+
type="password",
|
| 157 |
placeholder="๋น๋ฐ๋ฒํธ๋ฅผ ์
๋ ฅํ์ธ์"
|
| 158 |
)
|
| 159 |
|
| 160 |
+
# 2) ์
์ค์ผ์ผํ ์ด๋ฏธ์ง ์
๋ ฅ
|
| 161 |
input_image = gr.Image(label="์๋ณธ ์ด๋ฏธ์ง", type="pil")
|
|
|
|
| 162 |
|
| 163 |
+
# 3) ๊ณ ๊ธ ์ค์ ์์ฝ๋์ธ ์์ ํ๋ผ๋ฏธํฐ๋ค ์ ์
|
| 164 |
+
with gr.Accordion("๊ณ ๊ธ ์ค์ ", open=False):
|
| 165 |
+
scale_factor = gr.Slider(
|
| 166 |
+
minimum=1, maximum=4, value=2, step=0.5, label="ํ๋ ๋น์จ"
|
| 167 |
+
)
|
| 168 |
+
output_format = gr.Radio(
|
| 169 |
+
choices=["jpg", "png"], value="jpg", label="์ถ๋ ฅ ํ์"
|
| 170 |
+
)
|
| 171 |
+
sd_model = gr.Dropdown(
|
| 172 |
+
choices=[
|
| 173 |
+
"juggernaut_reborn.safetensors [338b85bc4f]",
|
| 174 |
+
"sd_xl_base_1.0.safetensors [39d4e625d]",
|
| 175 |
+
"sdxl_base_v0.9-9961.safetensors [3048045c]",
|
| 176 |
+
"realisticVisionV51_v51VAE.safetensors [5ecbfa0ac]",
|
| 177 |
+
],
|
| 178 |
+
value="juggernaut_reborn.safetensors [338b85bc4f]",
|
| 179 |
+
label="SD ๋ชจ๋ธ"
|
| 180 |
+
)
|
| 181 |
+
resemblance = gr.Slider(
|
| 182 |
+
minimum=0.0, maximum=1.0, value=0.6, step=0.05, label="์๋ณธ ์ ์ฌ๋"
|
| 183 |
+
)
|
| 184 |
+
creativity = gr.Slider(
|
| 185 |
+
minimum=0.0, maximum=1.0, value=0.35, step=0.05, label="์ฐฝ์์ฑ"
|
| 186 |
+
)
|
| 187 |
+
dynamic = gr.Slider(
|
| 188 |
+
minimum=1, maximum=10, value=6, step=1, label="๋ค์ด๋๋ฏน ์๊ณ๊ฐ"
|
| 189 |
+
)
|
| 190 |
+
sharpen = gr.Slider(
|
| 191 |
+
minimum=0, maximum=2, value=0, step=0.1, label="์ ๋ช
๋"
|
| 192 |
+
)
|
| 193 |
+
seed = gr.Number(
|
| 194 |
+
value=1337, label="์๋", precision=0
|
| 195 |
+
)
|
| 196 |
+
|
| 197 |
+
# 4) ์คํ ๋ฒํผ
|
| 198 |
submit_btn = gr.Button("์คํ", variant="primary")
|
| 199 |
|
| 200 |
with gr.Column():
|
| 201 |
+
# 5) ๊ฒฐ๊ณผ ์ด๋ฏธ์ง ํ์
|
| 202 |
output_image = gr.Image(label="๊ฐ์ ๋ ์ด๋ฏธ์ง")
|
| 203 |
|
| 204 |
+
# 6) ๋ฒํผ ํด๋ฆญ ์ ํธ์ถํ ํจ์
|
| 205 |
def process_image(
|
| 206 |
+
password, image, scale_factor, output_format, sd_model,
|
| 207 |
resemblance, creativity, dynamic, sharpen, seed
|
| 208 |
):
|
| 209 |
# ๋น๋ฐ๋ฒํธ ๊ฒ์ฆ
|
| 210 |
if password != "1089":
|
| 211 |
raise ValueError("์๋ชป๋ ๋น๋ฐ๋ฒํธ์
๋๋ค.")
|
|
|
|
| 212 |
if image is None:
|
| 213 |
return None
|
| 214 |
|
|
|
|
| 215 |
return upscale_image(
|
| 216 |
image=image,
|
| 217 |
scale_factor=scale_factor,
|
|
|
|
| 224 |
seed=int(seed)
|
| 225 |
)
|
| 226 |
|
| 227 |
+
# 7) click ์ด๋ฒคํธ์ ๋ชจ๋ ์
๋ ฅ ์์ ฏ์ ์์๋๋ก ์ ๋ฌ
|
| 228 |
submit_btn.click(
|
| 229 |
process_image,
|
| 230 |
inputs=[
|
| 231 |
+
password_input,
|
| 232 |
+
input_image, scale_factor, output_format, sd_model,
|
| 233 |
resemblance, creativity, dynamic, sharpen, seed
|
| 234 |
],
|
| 235 |
outputs=[output_image]
|