ohayonguy commited on
Commit
81ba96a
1 Parent(s): b1e0e58

improved interface

Browse files
Files changed (1) hide show
  1. app.py +10 -10
app.py CHANGED
@@ -148,9 +148,9 @@ def enhance_face(img, face_helper, has_aligned, num_flow_steps, scale=2):
148
  @torch.inference_mode()
149
  @spaces.GPU()
150
  def inference(
 
151
  seed,
152
  randomize_seed,
153
- img,
154
  aligned,
155
  scale,
156
  num_flow_steps,
@@ -250,16 +250,9 @@ demo = gr.Interface(
250
  inference,
251
  [
252
  gr.Image(label="Input", type="filepath", show_label=True),
 
253
  gr.Checkbox(label="Randomize seed", value=True),
254
  gr.Checkbox(label="The input is an aligned face image", value=False),
255
- gr.Slider(
256
- label="Number of Inference Steps",
257
- minimum=1,
258
- maximum=200,
259
- step=1,
260
- value=25,
261
- scale=1,
262
- ),
263
  gr.Slider(
264
  label="Scale factor (applicable to non-aligned face images)",
265
  minimum=1,
@@ -268,7 +261,14 @@ demo = gr.Interface(
268
  value=1,
269
  scale=1,
270
  ),
271
- gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, scale=1),
 
 
 
 
 
 
 
272
  ],
273
  [
274
  gr.Image(label="Output", type="numpy", show_label=True, format="png"),
 
148
  @torch.inference_mode()
149
  @spaces.GPU()
150
  def inference(
151
+ img,
152
  seed,
153
  randomize_seed,
 
154
  aligned,
155
  scale,
156
  num_flow_steps,
 
250
  inference,
251
  [
252
  gr.Image(label="Input", type="filepath", show_label=True),
253
+ gr.Slider(label="Seed", minimum=0, maximum=MAX_SEED, step=1, value=42, scale=1),
254
  gr.Checkbox(label="Randomize seed", value=True),
255
  gr.Checkbox(label="The input is an aligned face image", value=False),
 
 
 
 
 
 
 
 
256
  gr.Slider(
257
  label="Scale factor (applicable to non-aligned face images)",
258
  minimum=1,
 
261
  value=1,
262
  scale=1,
263
  ),
264
+ gr.Slider(
265
+ label="Number of Inference Steps",
266
+ minimum=1,
267
+ maximum=200,
268
+ step=1,
269
+ value=25,
270
+ scale=1,
271
+ ),
272
  ],
273
  [
274
  gr.Image(label="Output", type="numpy", show_label=True, format="png"),