Update app.py

app.py

@@ -64,7 +64,7 @@ def gen_pred(img, model=seg_model):
     img = tf.expand_dims(img, axis=0)
     pred = model.predict(img)
     pred = np.squeeze(pred, axis=0)
-    pil_img = Image.fromarray((x * 255).astype(np.uint8))
+    pil_img = Image.fromarray((pred * 255).astype(np.uint8))
     # pil_img = Image.fromarray(pred, 'RGB')
     # im = Image.fromarray((pred * 255).astype(np.uint8))
     # img_bytes = pred.tobytes()