Spaces:
Sleeping
Sleeping
Commit
·
b9eb1ac
1
Parent(s):
3d5d398
Update app.py
Browse files
app.py
CHANGED
@@ -73,7 +73,7 @@ def read_cells(cell,model):
|
|
73 |
# getting predictions and setting the values if probabilities are above 65%
|
74 |
|
75 |
predictions = model.predict(img)
|
76 |
-
classIndex = np.argmax(model.predict(
|
77 |
probabilityValue = np.amax(predictions)
|
78 |
|
79 |
if probabilityValue > 0.65:
|
|
|
73 |
# getting predictions and setting the values if probabilities are above 65%
|
74 |
|
75 |
predictions = model.predict(img)
|
76 |
+
classIndex = np.argmax(model.predict(img), axis=-1)
|
77 |
probabilityValue = np.amax(predictions)
|
78 |
|
79 |
if probabilityValue > 0.65:
|