ssboost commited on
Commit
ff5efd2
ยท
verified ยท
1 Parent(s): 0786ac5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +45 -12
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
- # 2) process_image ํ•จ์ˆ˜ ์‹œ๊ทธ๋‹ˆ์ฒ˜ ์ˆ˜์ •
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
- # 3) ํด๋ฆญ ์ด๋ฒคํŠธ์— password_input ์ถ”๊ฐ€
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]