vahidrezanezhad commited on
Commit
e7c39e9
·
verified ·
1 Parent(s): 437292f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -3
app.py CHANGED
@@ -22,9 +22,9 @@ def otsu_copy_binary(img):
22
 
23
  def visualize_model_output(prediction, img, model_name):
24
  if model_name == "SBB/eynollah-binarization":
25
- prediction_true = prediction_true * -1
26
- prediction_true = prediction_true + 1
27
- added_image = prediction_true * 255
28
  else:
29
  unique_classes = np.unique(prediction[:,:,0])
30
  rgb_colors = {'0' : [255, 255, 255],
 
22
 
23
  def visualize_model_output(prediction, img, model_name):
24
  if model_name == "SBB/eynollah-binarization":
25
+ prediction = prediction * -1
26
+ prediction = prediction + 1
27
+ added_image = prediction * 255
28
  else:
29
  unique_classes = np.unique(prediction[:,:,0])
30
  rgb_colors = {'0' : [255, 255, 255],