Update app.py

app.py

@@ -45,10 +45,21 @@ from tensorflow import keras
 from PIL import Image
 
 # Cell 1: Image Classification Model
-model1 = from_pretrained_keras("rocioadlc/Alejandra2")
-
-
-class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']
+model1 = from_pretrained_keras("rocioadlc/EfficientNetV2L")
+
+
+class_labels = ['battery',
+ 'biological',
+ 'brown-glass',
+ 'cardboard',
+ 'clothes',
+ 'green-glass',
+ 'metal',
+ 'paper',
+ 'plastic',
+ 'shoes',
+ 'trash',
+ 'white-glass']
 
 def predict_image(image_input):
     # Resize the image to the size expected by the model