Update app.py

app.py

@@ -3,6 +3,7 @@ import tensorflow as tf
 from tensorflow.keras.models import load_model
 from tensorflow.keras.preprocessing.image import img_to_array,load_img
 import numpy as np
+from PIL import Image
 import os
 
 # Load your model and tokenizer
@@ -20,7 +21,8 @@ model_path = r"inceptionv3_class.h5"
 hierarchical_models['class'] = load_model(model_path)
 
 def load_and_preprocess_image(image, target_size=(224, 224)):
-    image = load_img(image, target_size=(224, 224))
+    if isinstance(image, Image.Image):
+        image = image.resize(target_size)
     img_array = img_to_array(image)
     img_array = np.expand_dims(img_array, axis=0)
     img_array = tf.keras.applications.mobilenet_v2.preprocess_input(img_array)