Upload 3 files

.gitattributes

@@ -33,3 +33,4 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+plantvillage_efficientnet_b0.keras filter=lfs diff=lfs merge=lfs -text

app.py

@@ -0,0 +1,48 @@
+import gradio as gr
+import tensorflow as tf
+import numpy as np
+import os
+
+# 1. Load the model with error handling
+try:
+    model_path = 'plantvillage_efficientnet_b0.keras'
+    if not os.path.exists(model_path):
+        raise FileNotFoundError(f"Model file {model_path} not found in Space!")
+    
+    model = tf.keras.models.load_model(model_path, compile=False)
+    print("Model loaded successfully!")
+except Exception as e:
+    print(f"CRITICAL ERROR LOADING MODEL: {e}")
+    model = None
+
+classes = [
+    'Pepper__bell___Bacterial_spot', 'Pepper__bell___healthy', 'Potato___Early_blight',
+    'Potato___Late_blight', 'Potato___healthy', 'Tomato_Bacterial_spot',
+    'Tomato_Early_blight', 'Tomato_Late_blight', 'Tomato_Leaf_Mold',
+    'Tomato_Septoria_leaf_spot', 'Tomato_Spider_mites_Two_spotted_spider_mite',
+    'Tomato__Target_Spot', 'Tomato__Tomato_YellowLeaf__Curl_Virus',
+    'Tomato__Tomato_mosaic_virus', 'Tomato_healthy'
+]
+
+def predict(image):
+    if model is None:
+        return "Error: Model failed to load. Check Space logs."
+    
+    img = tf.cast(image, tf.float32)
+    img = tf.image.resize(img, (224, 224))
+    img = tf.keras.applications.efficientnet.preprocess_input(img)
+    img = tf.expand_dims(img, axis=0)
+
+    preds = model.predict(img)[0]
+    return {classes[i]: float(preds[i]) for i in range(len(classes))}
+
+demo = gr.Interface(
+    fn=predict,
+    inputs=gr.Image(),
+    outputs=gr.Label(num_top_classes=3),
+    title="Plant Disease Detector",
+    description="Identify plant leaf diseases."
+)
+
+if __name__ == '__main__':
+    demo.launch()

plantvillage_efficientnet_b0.keras

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:1db71d8fafcc5a0b0d7868dc3227310972017ab27ebefb7ebfa0d90e85e9bb18
+size 59863969

requirements.txt

@@ -0,0 +1,3 @@
+tensorflow
+gradio
+numpy