Update app.py

app.py

@@ -4,9 +4,14 @@ import tensorflow as tf
 
 models = [    {"name": "my_model_2.h5", "size": 512},    {"name": "my_model.h5", "size": 224},]
 
-def classify_image(image, model_name):
+from PIL import Image
+
+def classify_image(image_path, model_name):
     model_config = next(m for m in models if m["name"] == model_name)
     model = tf.keras.models.load_model(model_name)
+    image = Image.open(image_path).convert("RGB")
+    image = image.resize((model_config["size"], model_config["size"]))
+    image = np.array(image) / 255.0
     input_image = np.expand_dims(image, axis=0)
     prediction = model.predict(input_image).flatten()
     if len(prediction) > 1:
@@ -21,6 +26,7 @@ def classify_image(image, model_name):
         label = "Not glaucoma"
     return label, probability
 
+
 inputs = [
     gr.inputs.Image(shape=(224, 224), label="Eye image"),
     gr.inputs.Dropdown(choices=[m["name"] for m in models], label="Model"),