Update app.py

app.py

@@ -64,7 +64,8 @@ def style_transfer(input_image, artist):
     print("stylized_image:")
     print(stylized_image[0])
     print(stylized_image)
-    return tf.io.encode_jpeg(stylized_image[0])
+    #return tf.io.encode_jpeg(stylized_image[0])
+    return tf.keras.preprocessing.image.img_to_array(stylized_image[0])
     
 app = gr.Interface(
     style_transfer,