luca-martial commited on
Commit
d38a6c6
·
1 Parent(s): 7800a02

update examples

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -21,7 +21,8 @@ def stylize(content_image, style_image):
21
  stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
22
  return tensor_to_image(stylized_image)
23
 
24
- examples =[["example_paris.jpeg"], ["example_aristotle.jpeg"], ["example_dali.jpeg"], ["example_vangogh.jpeg"]]
 
25
  title = "Fast Neural Style Transfer using TF-Hub"
26
  description = "Demo for neural style transfer using the pretrained Arbitrary Image Stylization model from TensorFlow Hub."
27
 
@@ -31,5 +32,5 @@ iface = gr.Interface(fn=stylize,
31
  outputs="image",
32
  title=title,
33
  description=description,
34
- examples=examples)
35
  iface.launch()
 
21
  stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
22
  return tensor_to_image(stylized_image)
23
 
24
+ content_examples =[["example_paris.jpeg"], ["example_aristotle.jpeg"]]
25
+ style_examples = [["example_dali.jpeg"], ["example_vangogh.jpeg"]]
26
  title = "Fast Neural Style Transfer using TF-Hub"
27
  description = "Demo for neural style transfer using the pretrained Arbitrary Image Stylization model from TensorFlow Hub."
28
 
 
32
  outputs="image",
33
  title=title,
34
  description=description,
35
+ examples=[content_examples, style_examples])
36
  iface.launch()