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]
|