Update app.py

app.py

@@ -2,7 +2,7 @@ import gradio as gr
 import tensorflow as tf
 import numpy as np
 from tensorflow.keras.preprocessing import image
-from tensorflow.keras.applications.efficientnet_v2 import EfficientNetV2S, preprocess_input
+from tensorflow.keras.applications.efficientnet_v2 import preprocess_input
 from PIL import Image
 
 # Load the trained model
@@ -18,7 +18,7 @@ def preprocess_image(img: Image.Image) -> np.ndarray:
     """Preprocess the image to match the model's input requirements."""
     img = img.resize((224, 224))  # Resize image to model input size
     img_array = np.array(img)
-    img_array = EfficientNetV2S.preprocess_input(img_array)  # EfficientNetV2 preprocessing
+    img_array = preprocess_input(img_array)  # EfficientNetV2 preprocessing (correct usage)
     img_array = np.expand_dims(img_array, axis=0)  # Add batch dimension
     return img_array