lifedebugger commited on
Commit
eae52cb
·
verified ·
1 Parent(s): c81b5a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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', 'Food Organics', 'Glass', 'Metal', 'Miscellaneous Trash']
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)