Spaces:
Runtime error
Runtime error
enabled queue
Browse files
app.py
CHANGED
|
@@ -36,8 +36,8 @@ def fit_style_transfer(
|
|
| 36 |
epochs=4,
|
| 37 |
learning_rate=60.0,
|
| 38 |
steps_per_epoch=100,
|
| 39 |
-
style_weight=
|
| 40 |
-
content_weight=
|
| 41 |
var_weight=1e-12,
|
| 42 |
):
|
| 43 |
"""
|
|
@@ -200,7 +200,7 @@ def main():
|
|
| 200 |
style_weight = gr.Slider(
|
| 201 |
1e-4,
|
| 202 |
0.5,
|
| 203 |
-
value=
|
| 204 |
step=1e-4,
|
| 205 |
label="Style Weight",
|
| 206 |
)
|
|
@@ -208,15 +208,15 @@ def main():
|
|
| 208 |
content_weight = gr.Slider(
|
| 209 |
1e-3,
|
| 210 |
0.5,
|
| 211 |
-
value=0.
|
| 212 |
step=1e-4,
|
| 213 |
label="Content Weight",
|
| 214 |
)
|
| 215 |
|
| 216 |
var_weight = gr.Slider(
|
| 217 |
0,
|
| 218 |
-
1e-
|
| 219 |
-
value=1e-
|
| 220 |
step=1e-12,
|
| 221 |
label="Total Variation Weight",
|
| 222 |
)
|
|
@@ -245,9 +245,9 @@ def main():
|
|
| 245 |
4,
|
| 246 |
60,
|
| 247 |
100,
|
| 248 |
-
|
| 249 |
-
|
| 250 |
-
1e-
|
| 251 |
],
|
| 252 |
[
|
| 253 |
"examples/style_2.jpg",
|
|
@@ -258,9 +258,9 @@ def main():
|
|
| 258 |
4,
|
| 259 |
60,
|
| 260 |
100,
|
| 261 |
-
|
| 262 |
-
|
| 263 |
-
1e-
|
| 264 |
],
|
| 265 |
[
|
| 266 |
"examples/style_3.jpg",
|
|
@@ -271,9 +271,9 @@ def main():
|
|
| 271 |
4,
|
| 272 |
60,
|
| 273 |
100,
|
| 274 |
-
|
| 275 |
-
|
| 276 |
-
1e-
|
| 277 |
]
|
| 278 |
|
| 279 |
]
|
|
@@ -298,10 +298,10 @@ def main():
|
|
| 298 |
fn=fit_style_transfer,
|
| 299 |
inputs=inputs,
|
| 300 |
outputs=outputs,
|
| 301 |
-
examples=examples
|
|
|
|
| 302 |
)
|
| 303 |
|
| 304 |
-
interface.queue().launch(debug=True)
|
| 305 |
-
|
| 306 |
|
| 307 |
main()
|
|
|
|
| 36 |
epochs=4,
|
| 37 |
learning_rate=60.0,
|
| 38 |
steps_per_epoch=100,
|
| 39 |
+
style_weight=0.3,
|
| 40 |
+
content_weight=0.5,
|
| 41 |
var_weight=1e-12,
|
| 42 |
):
|
| 43 |
"""
|
|
|
|
| 200 |
style_weight = gr.Slider(
|
| 201 |
1e-4,
|
| 202 |
0.5,
|
| 203 |
+
value=0.3,
|
| 204 |
step=1e-4,
|
| 205 |
label="Style Weight",
|
| 206 |
)
|
|
|
|
| 208 |
content_weight = gr.Slider(
|
| 209 |
1e-3,
|
| 210 |
0.5,
|
| 211 |
+
value=0.5,
|
| 212 |
step=1e-4,
|
| 213 |
label="Content Weight",
|
| 214 |
)
|
| 215 |
|
| 216 |
var_weight = gr.Slider(
|
| 217 |
0,
|
| 218 |
+
1e-5,
|
| 219 |
+
value=1e-7,
|
| 220 |
step=1e-12,
|
| 221 |
label="Total Variation Weight",
|
| 222 |
)
|
|
|
|
| 245 |
4,
|
| 246 |
60,
|
| 247 |
100,
|
| 248 |
+
0.3,
|
| 249 |
+
0.5,
|
| 250 |
+
1e-8,
|
| 251 |
],
|
| 252 |
[
|
| 253 |
"examples/style_2.jpg",
|
|
|
|
| 258 |
4,
|
| 259 |
60,
|
| 260 |
100,
|
| 261 |
+
0.3,
|
| 262 |
+
0.5,
|
| 263 |
+
1e-5,
|
| 264 |
],
|
| 265 |
[
|
| 266 |
"examples/style_3.jpg",
|
|
|
|
| 271 |
4,
|
| 272 |
60,
|
| 273 |
100,
|
| 274 |
+
0.5,
|
| 275 |
+
0.3,
|
| 276 |
+
1e-10,
|
| 277 |
]
|
| 278 |
|
| 279 |
]
|
|
|
|
| 298 |
fn=fit_style_transfer,
|
| 299 |
inputs=inputs,
|
| 300 |
outputs=outputs,
|
| 301 |
+
examples=examples,
|
| 302 |
+
|
| 303 |
)
|
| 304 |
|
| 305 |
+
interface.queue().launch(debug=True, enable_queue=True)
|
|
|
|
| 306 |
|
| 307 |
main()
|