gurwindersingh commited on
Commit
43165fb
·
1 Parent(s): b75b567

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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 = model.predict_classes(img)
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(class_predictions, axis=1)
77
  probabilityValue = np.amax(predictions)
78
 
79
  if probabilityValue > 0.65: