Update app.py

app.py

@@ -15,10 +15,13 @@ def tensor_to_image(tensor):
         tensor = tensor[0]
     return Image.fromarray(tensor)
 
-def load(file_uploader):
+def load(file_uploader,fst=True):
     max_dim = 512
     image = file_uploader
-    image = tf.image.decode_image(image.read(), channels=3)
+    if fst:
+     image = tf.image.decode_image(image.read(), channels=3)
+    else: 
+        image = tf.image.decode_image(image, channels=3)
     image = tf.image.convert_image_dtype(image, tf.float32)
 
     shape = tf.cast(tf.shape(image)[:-1], tf.float32)
@@ -33,7 +36,7 @@ def load(file_uploader):
     
 st.title("Sketch to Real Image Pix2Pix")
 inp = st.file_uploader("Sketch input")
-style = load('style.jpg')
+style = load('style.jpg',False)
 
 if inp is not None:
     img = load(inp)