Spaces:
Runtime error
Runtime error
Giedrius Blazys
commited on
Commit
·
dcd9976
1
Parent(s):
416e479
switch probs
Browse files
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,
|
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)
|