luca-martial commited on
Commit
0da3808
1 Parent(s): d38a6c6
Files changed (1) hide show
  1. app.py +4 -3
app.py CHANGED
@@ -21,11 +21,12 @@ 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
- 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
-
 
29
 
30
  iface = gr.Interface(fn=stylize,
31
  inputs=["image", "image"],
 
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_vangogh.jpeg"]]
25
+ style_examples = [["example_aristotle.jpeg"], ["example_dali.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
+ inputs = gr.inputs.Image(label="Content Image", source=["upload", "webcam"])
29
+ outputs = gr.outputs.Image(label="Style Image")
30
 
31
  iface = gr.Interface(fn=stylize,
32
  inputs=["image", "image"],