luca-martial commited on
Commit
561e110
1 Parent(s): 0da3808
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -25,11 +25,11 @@ 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"],
33
  outputs="image",
34
  title=title,
35
  description=description,
 
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
+ content_input = gr.inputs.Image(label="Content Image", source=["upload", "webcam"])
29
+ style_input = gr.inputs.Image(label="Style Image", source=["upload", "webcam"])
30
 
31
  iface = gr.Interface(fn=stylize,
32
+ inputs=[content_input, style_input],
33
  outputs="image",
34
  title=title,
35
  description=description,