Update app.py

app.py

@@ -62,7 +62,6 @@ def predict_image(input):
     image_array = tf.expand_dims(image_array, 0)
     # Predict using the model
     predictions = model1.predict(image_array)
-    class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
     category_scores = {}
     for i, class_label in enumerate(class_labels):
         category_scores[class_label] = predictions[0][i].item()