luca-martial commited on
Commit
a80599f
1 Parent(s): 3c5019f
Files changed (1) hide show
  1. app.py +6 -4
app.py CHANGED
@@ -25,10 +25,12 @@ def load_img(path_to_img):
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)
 
25
 
26
 
27
  def stylize(content_path, style_path):
28
+ content_image = plt.imread(content_path)
29
+
30
+ #content_image = load_img(content_path)
31
+ #style_image = load_img(style_path)
32
+ #stylized_image = hub_model(tf.constant(content_image), tf.constant(style_image))[0]
33
+ return content_image
34
 
35
  #content_image = plt.imread(content_image_path)
36
  #style_image = plt.imread(style_image_path)