anodev commited on
Commit
e3698f4
·
verified ·
1 Parent(s): 2d99372

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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