Inoob commited on
Commit
4b490bc
·
verified ·
1 Parent(s): 420de8c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -178,7 +178,7 @@ net = build_model(hypar, device)
178
  def inference(image):
179
  image_path = image
180
 
181
- image_tensor, orig_size = cv2.imread(image_path)
182
  mask = predict(net, image_tensor)
183
 
184
 
 
178
  def inference(image):
179
  image_path = image
180
 
181
+ image_tensor = cv2.imread(image_path)
182
  mask = predict(net, image_tensor)
183
 
184