jcmc commited on
Commit
dd7783e
1 Parent(s): 4a8ae11

adjusting interface inputs

Browse files
Files changed (1) hide show
  1. app.py +12 -6
app.py CHANGED
@@ -165,7 +165,7 @@ def optimize_network(
165
  input_depth, 'skip',
166
  pad='reflection',
167
  skip_n33d=128, skip_n33u=128,
168
- skip_n11=4, num_scales=7, # If you decrease the output size to 256x256 you might want to use num_scales=6
169
  upsample_mode='bilinear',
170
  downsample_mode='lanczos2',
171
  )
@@ -272,8 +272,14 @@ def inference(
272
  return out, 'video.mp4'
273
 
274
  iface = gr.Interface(fn=inference,
275
- inputs=["number", "number", "number", "number", "number",
276
- gr.inputs.Checkbox(default=False, label="class_token"),
277
- gr.inputs.Checkbox(default=True, label="maximise"),
278
- "number"],
279
- outputs=["image","video"]).launch()
 
 
 
 
 
 
 
165
  input_depth, 'skip',
166
  pad='reflection',
167
  skip_n33d=128, skip_n33u=128,
168
+ skip_n11=4, num_scales=6, # If you decrease the output size to 256x256 you might want to use num_scales=6
169
  upsample_mode='bilinear',
170
  downsample_mode='lanczos2',
171
  )
 
272
  return out, 'video.mp4'
273
 
274
  iface = gr.Interface(fn=inference,
275
+ inputs=[
276
+ gr.inputs.Number(default=1e-3, label="learning rate"),
277
+ gr.inputs.Number(default=50, label="number of iterations (more is better)"),
278
+ gr.inputs.Number(default=32, label="cutn (number of cuts)"),
279
+ gr.inputs.Number(default=10, label="layer"),
280
+ gr.inputs.Number(default=1e-3, label="neuron"),
281
+ gr.inputs.Checkbox(default=False, label="class_token"),
282
+ gr.inputs.Checkbox(default=True, label="maximise"),
283
+ gr.inputs.Slider(minimum=0, maximum=30, default=10, label='display rate'),
284
+ ],
285
+ outputs=["image","video"]).launch()