SonFox2920 commited on
Commit
3dcabb5
·
verified ·
1 Parent(s): aed03ad

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -139,7 +139,7 @@ def main():
139
 
140
  # Make prediction
141
  prediction = model.predict(np.expand_dims(processed_image, axis=0))
142
- class_names = ['10', '7', '7.5', '8', '8.5', '9', '9.2', '9.5', '9.7']
143
 
144
  # Get top 5 predictions
145
  top_predictions = get_top_predictions(prediction, class_names)
 
139
 
140
  # Make prediction
141
  prediction = model.predict(np.expand_dims(processed_image, axis=0))
142
+ class_names = ['10', '6.5', '7', '7.5', '8', '8.5', '9', '9.2', '9.5', '9.7']
143
 
144
  # Get top 5 predictions
145
  top_predictions = get_top_predictions(prediction, class_names)