Upload folder using huggingface_hub

.gitattributes

@@ -1,35 +1,35 @@
-*.7z filter=lfs diff=lfs merge=lfs -text
-*.arrow filter=lfs diff=lfs merge=lfs -text
-*.bin filter=lfs diff=lfs merge=lfs -text
-*.bz2 filter=lfs diff=lfs merge=lfs -text
-*.ckpt filter=lfs diff=lfs merge=lfs -text
-*.ftz filter=lfs diff=lfs merge=lfs -text
-*.gz filter=lfs diff=lfs merge=lfs -text
-*.h5 filter=lfs diff=lfs merge=lfs -text
-*.joblib filter=lfs diff=lfs merge=lfs -text
-*.lfs.* filter=lfs diff=lfs merge=lfs -text
-*.mlmodel filter=lfs diff=lfs merge=lfs -text
-*.model filter=lfs diff=lfs merge=lfs -text
-*.msgpack filter=lfs diff=lfs merge=lfs -text
-*.npy filter=lfs diff=lfs merge=lfs -text
-*.npz filter=lfs diff=lfs merge=lfs -text
-*.onnx filter=lfs diff=lfs merge=lfs -text
-*.ot filter=lfs diff=lfs merge=lfs -text
-*.parquet filter=lfs diff=lfs merge=lfs -text
-*.pb filter=lfs diff=lfs merge=lfs -text
-*.pickle filter=lfs diff=lfs merge=lfs -text
-*.pkl filter=lfs diff=lfs merge=lfs -text
-*.pt filter=lfs diff=lfs merge=lfs -text
-*.pth filter=lfs diff=lfs merge=lfs -text
-*.rar filter=lfs diff=lfs merge=lfs -text
-*.safetensors filter=lfs diff=lfs merge=lfs -text
-saved_model/**/* filter=lfs diff=lfs merge=lfs -text
-*.tar.* filter=lfs diff=lfs merge=lfs -text
-*.tar filter=lfs diff=lfs merge=lfs -text
-*.tflite filter=lfs diff=lfs merge=lfs -text
-*.tgz filter=lfs diff=lfs merge=lfs -text
-*.wasm filter=lfs diff=lfs merge=lfs -text
-*.xz 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
+*.7z filter=lfs diff=lfs merge=lfs -text
+*.arrow filter=lfs diff=lfs merge=lfs -text
+*.bin filter=lfs diff=lfs merge=lfs -text
+*.bz2 filter=lfs diff=lfs merge=lfs -text
+*.ckpt filter=lfs diff=lfs merge=lfs -text
+*.ftz filter=lfs diff=lfs merge=lfs -text
+*.gz filter=lfs diff=lfs merge=lfs -text
+*.h5 filter=lfs diff=lfs merge=lfs -text
+*.joblib filter=lfs diff=lfs merge=lfs -text
+*.lfs.* filter=lfs diff=lfs merge=lfs -text
+*.mlmodel filter=lfs diff=lfs merge=lfs -text
+*.model filter=lfs diff=lfs merge=lfs -text
+*.msgpack filter=lfs diff=lfs merge=lfs -text
+*.npy filter=lfs diff=lfs merge=lfs -text
+*.npz filter=lfs diff=lfs merge=lfs -text
+*.onnx filter=lfs diff=lfs merge=lfs -text
+*.ot filter=lfs diff=lfs merge=lfs -text
+*.parquet filter=lfs diff=lfs merge=lfs -text
+*.pb filter=lfs diff=lfs merge=lfs -text
+*.pickle filter=lfs diff=lfs merge=lfs -text
+*.pkl filter=lfs diff=lfs merge=lfs -text
+*.pt filter=lfs diff=lfs merge=lfs -text
+*.pth filter=lfs diff=lfs merge=lfs -text
+*.rar filter=lfs diff=lfs merge=lfs -text
+*.safetensors filter=lfs diff=lfs merge=lfs -text
+saved_model/**/* filter=lfs diff=lfs merge=lfs -text
+*.tar.* filter=lfs diff=lfs merge=lfs -text
+*.tar filter=lfs diff=lfs merge=lfs -text
+*.tflite filter=lfs diff=lfs merge=lfs -text
+*.tgz filter=lfs diff=lfs merge=lfs -text
+*.wasm filter=lfs diff=lfs merge=lfs -text
+*.xz 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

Dockerfile

@@ -0,0 +1,17 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# copy code and requirements
+COPY . /app
+RUN pip install --upgrade pip
+RUN pip install --no-cache-dir -r requirements.txt
+
+# (optional) set env to tell uvicorn to use lower log/threads
+ENV PYTHONUNBUFFERED=1
+
+# Ports: Spaces expects the app to listen on 7860 by convention for UIs/APIs
+EXPOSE 7860
+
+# Start Uvicorn serving your FastAPI app module `app:app` (adjust if your module/file name differs)
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--loop", "uvloop", "--workers", "1"]

README.md

@@ -0,0 +1,10 @@
+---
+title: Crop Disease Detection
+emoji: 👀
+colorFrom: yellow
+colorTo: blue
+sdk: docker
+pinned: false
+---
+
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference

app.py

@@ -0,0 +1,220 @@
+import warnings
+import io
+from fastapi import FastAPI, UploadFile, File, HTTPException
+from pydantic import BaseModel
+from PIL import Image
+from fastai.learner import load_learner
+import joblib
+
+from models_download import download_models
+
+app = FastAPI()
+
+learn = None
+
+def read_imagefile(data):
+    return Image.open(io.BytesIO(data))
+
+
+def load_model_safely():
+    """Load the FastAI model with proper warning handling"""
+    try:
+        with warnings.catch_warnings():
+            warnings.filterwarnings("ignore", category=UserWarning, message=".*load_learner.*")
+            learn = load_learner("crop_disease_model.pkl")
+            return learn
+    except FileNotFoundError:
+        print("Warning: Model file 'crop_disease_model.pkl' not found. Please ensure the model file exists.")
+        return None
+    except Exception as e:
+        print(f"Error loading model: {e}")
+        return None
+
+disease_info = {
+    "Wheat__Healthy": {
+        "crop": "Wheat",
+        "description": "Healthy wheat plant without any visible disease symptoms.",
+        "symptoms": [],
+        "treatment": [],
+        "prevention": [
+            "Maintain balanced soil fertility",
+            "Ensure proper irrigation",
+            "Use certified disease-free seeds"
+        ]
+    },
+    "Wheat__Brown_Rust": {
+        "crop": "Wheat",
+        "description": "Brown rust is caused by Puccinia triticina, a major foliar disease of wheat.",
+        "symptoms": [
+            "Small, circular to oval brown pustules on leaves",
+            "Yellowing around lesions",
+            "Reduced grain yield and quality"
+        ],
+        "treatment": [
+            "Apply fungicides such as Propiconazole or Tebuconazole",
+            "Early spray during initial infection"
+        ],
+        "prevention": [
+            "Plant resistant wheat varieties",
+            "Destroy volunteer wheat plants (green bridge)",
+            "Rotate crops to break pathogen cycle"
+        ]
+    },
+
+    "Rice__Healthy": {
+        "crop": "Rice",
+        "description": "Healthy rice plant with no disease or insect damage.",
+        "symptoms": [],
+        "treatment": [],
+        "prevention": [
+            "Use disease-free seeds",
+            "Ensure balanced fertilizer application",
+            "Maintain proper water management"
+        ]
+    },
+    "Rice__Leaf_Blast": {
+        "crop": "Rice",
+        "description": "Leaf blast caused by Magnaporthe oryzae affects rice leaves.",
+        "symptoms": [
+            "Diamond-shaped lesions on leaves",
+            "Grayish centers with brown borders",
+            "Rapid drying of leaves in severe cases"
+        ],
+        "treatment": [
+            "Apply fungicides such as Tricyclazole or Isoprothiolane",
+            "Remove infected plants to reduce spread"
+        ],
+        "prevention": [
+            "Grow resistant rice varieties",
+            "Avoid excess nitrogen fertilizer",
+            "Ensure proper spacing to reduce humidity"
+        ]
+    },
+    "Rice__Neck_Blast": {
+        "crop": "Rice",
+        "description": "Neck blast affects the panicle neck, caused by Magnaporthe oryzae.",
+        "symptoms": [
+            "Blackening and rotting at panicle neck",
+            "Panicles dry prematurely",
+            "Grain filling is reduced or absent"
+        ],
+        "treatment": [
+            "Spray fungicides at heading stage (Tricyclazole, Carbendazim)",
+            "Destroy infected residues"
+        ],
+        "prevention": [
+            "Cultivate resistant varieties",
+            "Balanced fertilizer use",
+            "Crop rotation with non-host crops"
+        ]
+    },
+
+    "Potato__Healthy": {
+        "crop": "Potato",
+        "description": "Healthy potato leaf without disease or pest damage.",
+        "symptoms": [],
+        "treatment": [],
+        "prevention": [
+            "Plant certified seed potatoes",
+            "Follow crop rotation",
+            "Maintain field hygiene"
+        ]
+    },
+    "Potato__Late_Blight": {
+        "crop": "Potato",
+        "description": "Late blight, caused by Phytophthora infestans, is a devastating potato disease.",
+        "symptoms": [
+            "Dark brown to black lesions on leaves with pale green halos",
+            "White fungal growth under moist conditions",
+            "Tuber rot in severe cases"
+        ],
+        "treatment": [
+            "Apply fungicides such as Mancozeb or Metalaxyl",
+            "Remove and destroy infected plants"
+        ],
+        "prevention": [
+            "Plant resistant potato varieties",
+            "Avoid overhead irrigation",
+            "Ensure proper crop rotation"
+        ]
+    },
+    "Potato__Early_Blight": {
+        "crop": "Potato",
+        "description": "Early blight, caused by Alternaria solani, is a common fungal disease of potato.",
+        "symptoms": [
+            "Small dark brown spots with concentric rings on leaves",
+            "Premature leaf drop",
+            "Reduced tuber yield"
+        ],
+        "treatment": [
+            "Apply fungicides like Chlorothalonil or Copper-based sprays",
+            "Remove infected leaves early"
+        ],
+        "prevention": [
+            "Use resistant varieties",
+            "Practice crop rotation",
+            "Avoid excessive nitrogen fertilizer"
+        ]
+    }
+}
+
+
+@app.on_event("startup")
+def startup_event():
+    global learn
+    download_models()
+    learn = load_model_safely()
+
+
+@app.get("/")
+def read_root():
+    return {
+        "message": "Welcome to AgroAI - Crop Disease Detection API",
+        "status": "running",
+        "model_loaded": learn is not None
+    }
+
+
+@app.post("/predict/")
+async def predict(file: UploadFile = File(...)):
+    if learn is None:
+        raise HTTPException(
+            status_code=503,
+            detail="Model not available. Please check server configuration."
+        )
+
+    if not file.content_type or not file.content_type.startswith("image/"):
+        raise HTTPException(
+            status_code=400,
+            detail="File must be an image"
+        )
+
+    try:
+        # Read and process the image
+        image_data = await file.read()
+        img = read_imagefile(image_data)
+
+        # Make prediction
+        pred, pred_idx, probs = learn.predict(img)
+        pred_str = str(pred)
+
+        info = disease_info.get(pred_str, {
+            "crop": "",
+            "description": "",
+            "symptoms": [pred, pred_idx],
+            "treatment": [],
+            "prevention": []
+        })
+        0
+        return {
+            "prediction": pred_str,
+            "confidence": float(probs[pred_idx]),
+            "filename": file.filename,
+            **info
+        }
+
+    except Exception as e:
+        raise HTTPException(
+            status_code=500,
+            detail=f"Error processing image: {str(e)}"
+        )

crop_disease_model.pkl

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

models_download.py

@@ -0,0 +1,17 @@
+from huggingface_hub import hf_hub_download
+import os
+
+def download_models():
+
+    files = [
+        ("Bhavesh003/crop-disease-model", "crop_disease_model.pkl")
+    ]
+
+    for repo, fname in files:
+        if not os.path.exists(fname):
+            hf_hub_download(
+                repo_id=repo,
+                filename=fname,
+                local_dir=".",
+                force_filename=fname
+            )

requirements.txt

@@ -0,0 +1,10 @@
+fastapi
+uvicorn[standard]
+pydantic
+joblib
+scikit-learn
+fastai
+torch
+pillow
+huggingface-hub
+python-multipart