Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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)
|