Spaces:
Running
Running
fix-polygon-contour-always-green
#4
by
NBoukachab
- opened
app.py
CHANGED
@@ -67,7 +67,7 @@ def query_image(model_name: gr.Dropdown, image: gr.Image) -> list([Image, json])
|
|
67 |
|
68 |
# Make a prediction with the model
|
69 |
detected_polygons, probabilities, mask, overlap = ufcn_model.model.predict(
|
70 |
-
input_image=image, raw_output=True, mask_output=
|
71 |
)
|
72 |
|
73 |
# Load image
|
|
|
67 |
|
68 |
# Make a prediction with the model
|
69 |
detected_polygons, probabilities, mask, overlap = ufcn_model.model.predict(
|
70 |
+
input_image=image, raw_output=True, mask_output=False, overlap_output=False
|
71 |
)
|
72 |
|
73 |
# Load image
|