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