Update app.py

app.py

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