Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -178,7 +178,7 @@ net = build_model(hypar, device)
|
|
178 |
def inference(image):
|
179 |
image_path = image
|
180 |
|
181 |
-
image_tensor
|
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 |
|