breynolds1247 commited on
Commit
077e5fa
1 Parent(s): f1869c7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -61,7 +61,10 @@ def style_transfer(input_image, artist):
61
  #Pass content and style images as arguments in TensorFlow Constant object format
62
  stylized_image = hub_module(tf.constant(content_image), tf.constant(style_image))[0]
63
 
64
- return stylized_image
 
 
 
65
 
66
  app = gr.Interface(
67
  style_transfer,
 
61
  #Pass content and style images as arguments in TensorFlow Constant object format
62
  stylized_image = hub_module(tf.constant(content_image), tf.constant(style_image))[0]
63
 
64
+ print("stylized_image:")
65
+ print(stylized_image[0])
66
+ print(stylized_image)
67
+ return stylized_image[0]
68
 
69
  app = gr.Interface(
70
  style_transfer,