Spaces:
Runtime error
Runtime error
adjusting interface inputs
Browse files
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=
|
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 |
-
|
276 |
-
|
277 |
-
|
278 |
-
|
279 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
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()
|