mustapha commited on
Commit
2659416
1 Parent(s): ce57e72

Saliency map

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -85,8 +85,8 @@ def segment(image):
85
  output = model.predict(image)
86
  # print(output)
87
  labels = {
88
- "farsi" : 1-float(output),
89
- "ruqaa" : float(output)
90
  }
91
  grads, _ = saliency_map(image[0, :, :, 0])
92
  s_map = grads.numpy()[0, :, :, 0]
 
85
  output = model.predict(image)
86
  # print(output)
87
  labels = {
88
+ "Farsi" : 1-float(output),
89
+ "Ruqaa" : float(output)
90
  }
91
  grads, _ = saliency_map(image[0, :, :, 0])
92
  s_map = grads.numpy()[0, :, :, 0]