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()
|