Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def predict_image(img):
|
|
| 22 |
predictions = model_keenam.predict(img_array)
|
| 23 |
|
| 24 |
# Mendapatkan indeks kelas dengan probabilitas tertinggi
|
| 25 |
-
class_names = ['Cardboard',
|
| 26 |
predicted_class = class_names[np.argmax(predictions)]
|
| 27 |
|
| 28 |
return predicted_class, np.max(predictions)
|
|
|
|
| 22 |
predictions = model_keenam.predict(img_array)
|
| 23 |
|
| 24 |
# Mendapatkan indeks kelas dengan probabilitas tertinggi
|
| 25 |
+
class_names = ['Cardboard','Food Organics','Glass','Metal','Miscellaneous Trash','Paper','Plastic','Textile Trash','Vegetation']
|
| 26 |
predicted_class = class_names[np.argmax(predictions)]
|
| 27 |
|
| 28 |
return predicted_class, np.max(predictions)
|