Giedrius Blazys commited on
Commit
dcd9976
1 Parent(s): 416e479

switch probs

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -13,7 +13,7 @@ def classify_image(img):
13
  img_array_expanded_dims = img.reshape((-1, 128, 128, 3))
14
  prediction = model.predict(img_array_expanded_dims)
15
  prediction_prob = float(tf.nn.sigmoid(prediction))
16
- probs = [prediction_prob, 1-prediction_prob]
17
  return dict(zip(categories, probs))
18
 
19
  gr_image = gr.inputs.Image(shape=IMG_SIZE)
 
13
  img_array_expanded_dims = img.reshape((-1, 128, 128, 3))
14
  prediction = model.predict(img_array_expanded_dims)
15
  prediction_prob = float(tf.nn.sigmoid(prediction))
16
+ probs = [1-prediction_prob, prediction_prob]
17
  return dict(zip(categories, probs))
18
 
19
  gr_image = gr.inputs.Image(shape=IMG_SIZE)