Spaces:
Running
Running
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 |
|