Update app.py

app.py

@@ -210,7 +210,7 @@ def sepia(input_img):
     img_batch = np.expand_dims(img, axis=0)
     img_batch = img_batch.astype(float)
     
-    outputs = sess.run(None, {"pixel_values": input_img})[0]    
+    outputs = sess.run(None, {"pixel_values": img_batch})[0]    
     logits = outputs.logits
 
     logits = tf.transpose(logits, [0, 2, 3, 1])