Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -95,9 +95,9 @@ def predict(jpg, msk):
|
|
95 |
model = onnxruntime.InferenceSession('lama.onnx', sess_options=sess_options)
|
96 |
|
97 |
image = Image.open(jpg)
|
98 |
-
mask = Image.open(msk).convert("L")
|
99 |
|
100 |
-
image, mask = prepare_img_and_mask(image.resize((512, 512)), mask.resize((512, 512), 'cpu')
|
101 |
# Run the model
|
102 |
outputs = model.run(None, {'l_image_': image.numpy().astype(np.float32), 'l_mask_': mask.numpy().astype(np.float32)})
|
103 |
|
|
|
95 |
model = onnxruntime.InferenceSession('lama.onnx', sess_options=sess_options)
|
96 |
|
97 |
image = Image.open(jpg)
|
98 |
+
mask = Image.open(msk).convert("L")
|
99 |
|
100 |
+
image, mask = prepare_img_and_mask(image.resize((512, 512)), mask.resize((512, 512)), 'cpu')
|
101 |
# Run the model
|
102 |
outputs = model.run(None, {'l_image_': image.numpy().astype(np.float32), 'l_mask_': mask.numpy().astype(np.float32)})
|
103 |
|