Update app.py

app.py

@@ -206,9 +206,10 @@ def draw_plot(pred_img, seg):
     return fig
 
 def sepia(input_img):
-    input_img = Image.fromarray(input_img)
-    outputs = sess.run(None, {"pixel_values": input_img})
+    img = Image.fromarray(input_img)
+    img_batch = np.expand_dims(img, axis=0)
     
+    outputs = sess.run(None, {"pixel_values": input_img})[0]    
     logits = outputs.logits
 
     logits = tf.transpose(logits, [0, 2, 3, 1])