Update app.py

app.py

@@ -99,7 +99,7 @@ def predict(jpg, msk):
 
     image, mask = prepare_img_and_mask(imagex.resize((512, 512)), mask.resize((512, 512)), 'cpu')
     # Run the model
-    outputs = model.run(None, {'l_image_': image.numpy().astype(np.float32), 'l_mask_': mask.numpy().astype(np.float32)})
+    outputs = model.run(None, {'image': image.numpy().astype(np.float32), 'mask': mask.numpy().astype(np.float32)})
 
     output = outputs[0][0]
     # Postprocess the outputs