bauckluc commited on
Commit
105ccac
·
verified ·
1 Parent(s): 5a1ed07

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -40,7 +40,7 @@ def predict_bmwX(image):
40
  return f"Error: Number of model outputs ({len(prediction[0])}) does not match number of class names ({len(class_names)})."
41
 
42
  # Apply threshold and set probabilities lower than 0.015 to 0.0
43
- threshold = 0.015
44
  prediction = np.array(prediction)
45
  prediction[prediction < threshold] = 0.0
46
 
 
40
  return f"Error: Number of model outputs ({len(prediction[0])}) does not match number of class names ({len(class_names)})."
41
 
42
  # Apply threshold and set probabilities lower than 0.015 to 0.0
43
+ threshold = 0.01395
44
  prediction = np.array(prediction)
45
  prediction[prediction < threshold] = 0.0
46