Change model

app.py

@@ -7,7 +7,7 @@ from PIL import Image
 from datasets import load_dataset
 from huggingface_hub import hf_hub_download
 import tensorflow as tf
-from tensorflow.keras.applications.resnet50 import preprocess_input
+import tensorflow as tf
 
 # --- LOAD CLASS LABELS FROM DATASET ---
 ds = load_dataset("dvk65/TrashTypes")
@@ -19,24 +19,18 @@ class_names = ds["train"].features["label"].names
 # --- LOAD MODEL ---
 REPO_ID = "dvk65/trash-classifier-resnet50"
 FILENAME = "trashclassify_13.keras"
-
 model_path = hf_hub_download(repo_id=REPO_ID, filename=FILENAME)
 
-model = tf.keras.models.load_model(
-    model_path,
-    custom_objects={"preprocess_input": preprocess_input}
-)
+model = tf.keras.models.load_model(model_path)
 
 
 def preprocess(image):
     image = image.resize((224, 224))
-    image = np.array(image)
-    image = preprocess_input(image)
+    image = np.array(image).astype("float32")
     image = np.expand_dims(image, axis=0)
     return image
 
 def predict(img):
-
     img = preprocess(img)
     preds = model.predict(img)[0]
     return {class_names[i]: float(preds[i]) for i in range(len(preds))}