fix-polygon-contour-always-green

#4
by NBoukachab - opened
Files changed (1) hide show
  1. app.py +1 -1
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=True, overlap_output=True
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