Update app.py

app.py

@@ -208,6 +208,7 @@ def draw_plot(pred_img, seg):
 def sepia(input_img):
     img = Image.fromarray(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]    
     logits = outputs.logits