Update app.py

app.py

@@ -209,6 +209,7 @@ def sepia(input_img):
     img = cv2.imread(input_img).astype(np.float32)
     img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
     img_batch = np.expand_dims(img, axis=0)
+    img_batch = np.transpose(img_batch, (0, 3, 1, 2))
     
     outputs = sess.run(None, {"pixel_values": img_batch})[0]    
     logits = outputs.logits