debug

app.py

@@ -1,11 +1,19 @@
 import gradio as gr
 import tensorflow as tf
 import tensorflow_hub as hub
-import matplotlib.pylab as plt
+import matplotlib.pyplot as plt
 import numpy as np
 
 hub_model = hub.load('https://tfhub.dev/google/magenta/arbitrary-image-stylization-v1-256/2')
 
+def tensor_to_image(tensor):
+  tensor = tensor*255
+  tensor = np.array(tensor, dtype=np.uint8)
+  if np.ndim(tensor)>3:
+    assert tensor.shape[0] == 1
+    tensor = tensor[0]
+  return PIL.Image.fromarray(tensor)
+
 def load_img(path_to_img):
     max_dim = 512
     #img = tf.io.read_file(path_to_img)
@@ -24,8 +32,9 @@ def load_img(path_to_img):
   
 
 
-def stylize(content_path, style_path):
-    content_image = plt.imread(content_path)
+def stylize(content_image, style_image):
+    content_image = load_img(content_image)
+    content_image = tensor_to_image(content_image)
 
     #content_image = load_img(content_path)
     #style_image = load_img(style_path)