anodev commited on
Commit
b134327
·
verified ·
1 Parent(s): b2fc897

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
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
- sess_options = onnxruntime.SessionOptions()
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")