keras mod

app.py

@@ -37,7 +37,17 @@ custom_title = "<span style='color: rgb(243, 239, 224);'>Green Greta</span>"
 
 
 # Cell 1: Image Classification Model
-model1 = from_pretrained_keras("rocioadlc/efficientnetB0_trash")
+
+# model1 = from_pretrained_keras("rocioadlc/efficientnetB0_trash")
+
+model1 = tf.keras.layers.TFSMLayer(
+   'rocioadlc/efficientnetB0_trash',
+    call_endpoint='serving_default',
+    call_training_endpoint=None,
+    trainable=True,
+    name=None,
+    dtype=None
+)
 
 # Define class labels
 class_labels = ['cardboard', 'glass', 'metal', 'paper', 'plastic', 'trash']