luca-martial commited on
Commit
3c5019f
1 Parent(s): affa35b
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -24,11 +24,11 @@ def load_img(path_to_img):
24
 
25
 
26
 
27
- def stylize(content_path, style_image):
28
  content_image = load_img(content_path)
29
-
30
-
31
- return content_image
32
 
33
  #content_image = plt.imread(content_image_path)
34
  #style_image = plt.imread(style_image_path)
 
24
 
25
 
26
 
27
+ def stylize(content_path, style_path):
28
  content_image = load_img(content_path)
29
+ style_image = load_img(style_path)
30
+ stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
31
+ return tensor_to_image(content_image)
32
 
33
  #content_image = plt.imread(content_image_path)
34
  #style_image = plt.imread(style_image_path)