Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -16,7 +16,8 @@ import numpy as np
|
|
16 |
import onnxruntime
|
17 |
import torch
|
18 |
from PIL import Image
|
19 |
-
|
|
|
20 |
|
21 |
# Source https://github.com/advimman/lama
|
22 |
def get_image(image):
|
@@ -91,8 +92,7 @@ def prepare_img_and_mask(image, mask, device, pad_out_to_modulo=8, scale_factor=
|
|
91 |
|
92 |
|
93 |
def predict(jpg, msk):
|
94 |
-
|
95 |
-
model = onnxruntime.InferenceSession('lama.onnx', sess_options=sess_options)
|
96 |
|
97 |
imagex = Image.open(jpg)
|
98 |
mask = Image.open(msk).convert("L")
|
|
|
16 |
import onnxruntime
|
17 |
import torch
|
18 |
from PIL import Image
|
19 |
+
sess_options = onnxruntime.SessionOptions()
|
20 |
+
model = onnxruntime.InferenceSession('lama_fp32.onnx', sess_options=sess_options)
|
21 |
|
22 |
# Source https://github.com/advimman/lama
|
23 |
def get_image(image):
|
|
|
92 |
|
93 |
|
94 |
def predict(jpg, msk):
|
95 |
+
|
|
|
96 |
|
97 |
imagex = Image.open(jpg)
|
98 |
mask = Image.open(msk).convert("L")
|