fbadine commited on
Commit
02ff9a3
1 Parent(s): 493d10c

Update app.py

Browse files

Fixed a bug related to prediction being np.float32 not float

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -26,7 +26,7 @@ def predict(image):
26
 
27
  # Predict
28
  pred = model.predict(resized_image)
29
- prob = pred[0][0]
30
 
31
  scoring_output = {
32
  "Spam": prob,
 
26
 
27
  # Predict
28
  pred = model.predict(resized_image)
29
+ prob = float(pred[0][0])
30
 
31
  scoring_output = {
32
  "Spam": prob,