Update app.py

app.py

@@ -94,10 +94,10 @@ def predict(jpg, msk):
     sess_options = onnxruntime.SessionOptions()
     model = onnxruntime.InferenceSession('lama.onnx', sess_options=sess_options)
 
-    image = Image.open(jpg).resize((512, 512))
-    mask = Image.open(msk).convert("L").resize((512, 512))
+    image = Image.open(jpg)
+    mask = Image.open(msk).convert("L"))
 
-    image, mask = prepare_img_and_mask(image, mask, 'cpu')
+    image, mask = prepare_img_and_mask(image.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)})
 
@@ -106,6 +106,7 @@ def predict(jpg, msk):
     output = output.transpose(1, 2, 0)
     output = output.astype(np.uint8)
     output = Image.fromarray(output)
+    output = output.resize(image.size)
     output.save("/home/user/app/dataout/data_mask.png")