Create app.py

app.py

@@ -0,0 +1,66 @@
+import gradio as gr
+from transformers import pipeline
+from PIL import Image
+
+# Initialize the plant disease classification pipeline
+# You can replace the model with any fine-tuned plant disease model hosted on Hugging Face
+plant_disease_classifier = pipeline(
+    task="image-classification",  
+    model="nielsr/plant-disease-model",  
+    top_k=3  # return top 3 predictions
+)
+
+def diagnose_plant_health(image: Image.Image):
+    """
+    Takes a PIL Image of a plant leaf and returns:
+      - Top predicted disease label
+      - Confidence score
+      - Care advice based on the label
+    """
+    # Run the image through the classification pipeline
+    results = plant_disease_classifier(image)
+
+    # Format top-3 predictions
+    predictions = []
+    for res in results:
+        label = res['label']
+        score = res['score']
+        predictions.append(f"{label} ({score*100:.1f}%)")
+
+    # Determine advice based on the top prediction
+    top_label = results[0]['label'].lower()
+    if "healthy" in top_label:
+        advice = "Your plant looks healthy! Maintain regular watering and adequate sunlight."
+    else:
+        advice = (
+            f"Detected symptom: {results[0]['label']}. "
+            "Consider the following care steps:\n"
+            "1. Isolate the plant to prevent spread.\n"
+            "2. Prune affected areas with sterilized tools.\n"
+            "3. Apply an appropriate fungicide or treatment."
+        )
+
+    return "\n".join(predictions), advice
+
+# Building the Gradio interface
+iface = gr.Interface(
+    fn=diagnose_plant_health,
+    inputs=gr.Image(type="pil", label="Upload Plant Leaf Image"),
+    outputs=[
+        gr.Textbox(label="Predicted Diseases (Top 3)"),
+        gr.Textbox(label="Care Advice")
+    ],
+    title="Home Plant Health Monitor",
+    description=(
+        "Upload a photo of your plant's leaf to diagnose diseases and receive care recommendations. "
+        "This app uses a fine-tuned image-classification model on common plant diseases."
+    ),
+    examples=[
+        ["example_images/leaf_healthy.jpg"],
+        ["example_images/leaf_spot.jpg"]
+    ],
+    allow_flagging="never"
+)
+
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", server_port=7860)