Franco Astegiano commited on
Commit
617cb3b
·
unverified ·
1 Parent(s): 06e9b24

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -180,10 +180,10 @@ def modify(imageinput,style_input):
180
 
181
  # Gradio app
182
 
183
-
184
  app_interface = gr.Interface(modify,
185
  inputs=[content_image_input, style_image_input],
186
- outputs=outputs[0],
187
  title="Fast Neural Style Transfer",
188
  description="Gradio demo for Fast Neural Style Transfer using a pretrained Image Stylization model from TensorFlow Hub. To use it, simply upload a content image and style image, or click one of the examples to load them. To learn more about the project, please find the references listed below.",
189
  )
 
180
 
181
  # Gradio app
182
 
183
+ label = gr.output.image(modify(content_image_input, style_image_input))
184
  app_interface = gr.Interface(modify,
185
  inputs=[content_image_input, style_image_input],
186
+ outputs=label,
187
  title="Fast Neural Style Transfer",
188
  description="Gradio demo for Fast Neural Style Transfer using a pretrained Image Stylization model from TensorFlow Hub. To use it, simply upload a content image and style image, or click one of the examples to load them. To learn more about the project, please find the references listed below.",
189
  )