debug

my_app.py

@@ -23,14 +23,19 @@ This model is based on the EfficientNetB0 architecture and was trained on augmen
 """
 
 def preprocess(image):
+    print("before resize", image.shape)
     image = tf.image.resize(image, [224, 224])
+    
     image = tf.expand_dims(image, axis=0)
+    print("After expanddims", image.shape)
     return image
 
 def predict(model_selection, image):
     # Choose the model based on the dropdown selection
     model = model1 if model_selection == "EfficentNetB0 Fine Tune" else model2
 
+    print(model.summary())
+
     image = preprocess(image)
     prediction = model.predict(image)
     print("model prediction", prediction)