SKT27182 commited on
Commit
f795b54
1 Parent(s): 280a2a5

enabled queue

Browse files
Files changed (1) hide show
  1. app.py +18 -18
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=1e-2,
40
- content_weight=1e-4,
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=1e-1,
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.3,
212
  step=1e-4,
213
  label="Content Weight",
214
  )
215
 
216
  var_weight = gr.Slider(
217
  0,
218
- 1e-1,
219
- value=1e-5,
220
  step=1e-12,
221
  label="Total Variation Weight",
222
  )
@@ -245,9 +245,9 @@ def main():
245
  4,
246
  60,
247
  100,
248
- 1e-2,
249
- 1e-2,
250
- 1e-11,
251
  ],
252
  [
253
  "examples/style_2.jpg",
@@ -258,9 +258,9 @@ def main():
258
  4,
259
  60,
260
  100,
261
- 1e-2,
262
- 1e-2,
263
- 1e-11,
264
  ],
265
  [
266
  "examples/style_3.jpg",
@@ -271,9 +271,9 @@ def main():
271
  4,
272
  60,
273
  100,
274
- 1e-2,
275
- 1e-2,
276
- 1e-11,
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()