Upload 4 files

app1.py

@@ -0,0 +1,50 @@
+from fastapi import FastAPI, File, UploadFile, HTTPException
+from fastapi.responses import JSONResponse
+import tensorflow as tf
+import numpy as np
+from PIL import Image
+import io
+import json
+
+app = FastAPI()
+
+# Load the TensorFlow model
+model = tf.keras.models.load_model('./plant_disease_detection.h5')
+
+# Load categories
+with open('./categories.json') as f:
+    categories = json.load(f)
+
+def preprocess_image(image_bytes):
+    # Convert the image to a NumPy array
+    image = Image.open(io.BytesIO(image_bytes))
+    image = image.resize((224, 224))  # Adjust size as needed
+    image_array = np.array(image) / 255.0  # Normalize to [0, 1]
+    image_array = np.expand_dims(image_array, axis=0)  # Add batch dimension
+    return image_array
+
+@app.post('/predict')
+async def predict(file: UploadFile = File(...)):
+    if file.content_type.startswith('image/') is False:
+        raise HTTPException(status_code=400, detail='Invalid file type')
+
+    image_bytes = await file.read()
+    image_array = preprocess_image(image_bytes)
+
+    # Make prediction
+    predictions = model.predict(image_array)
+    predicted_class = np.argmax(predictions, axis=1)[0]
+
+    # Map to category names
+    predicted_label = categories.get(str(predicted_class), 'Unknown')
+
+    return JSONResponse(content={
+        'class': predicted_label,
+        'confidence': float(predictions[0][predicted_class])
+    })
+
+
+if __name__ == '__main__':
+    import uvicorn
+    uvicorn.run(app, host='0.0.0.0', port=8080)
+

categories.json

@@ -0,0 +1,40 @@
+{
+    "0": "Apple___Apple_scab",
+    "1": "Apple___Black_rot",
+    "2": "Apple___Cedar_apple_rust",
+    "3": "Apple___healthy",
+    "4": "Blueberry___healthy",
+    "5": "Cherry_(including_sour)___Powdery_mildew",
+    "6": "Cherry_(including_sour)___healthy",
+    "7": "Corn_(maize)___Cercospora_leaf_spot Gray_leaf_spot",
+    "8": "Corn_(maize)___Common_rust_",
+    "9": "Corn_(maize)___Northern_Leaf_Blight",
+    "10": "Corn_(maize)___healthy",
+    "11": "Grape___Black_rot",
+    "12": "Grape___Esca_(Black_Measles)",
+    "13": "Grape___Leaf_blight_(Isariopsis_Leaf_Spot)",
+    "14": "Grape___healthy",
+    "15": "Orange___Haunglongbing_(Citrus_greening)",
+    "16": "Peach___Bacterial_spot",
+    "17": "Peach___healthy",
+    "18": "Pepper,_bell___Bacterial_spot",
+    "19": "Pepper,_bell___healthy",
+    "20": "Potato___Early_blight",
+    "21": "Potato___Late_blight",
+    "22": "Potato___healthy",
+    "23": "Raspberry___healthy",
+    "24": "Soybean___healthy",
+    "25": "Squash___Powdery_mildew",
+    "26": "Strawberry___Leaf_scorch",
+    "27": "Strawberry___healthy",
+    "28": "Tomato___Bacterial_spot",
+    "29": "Tomato___Early_blight",
+    "30": "Tomato___Late_blight",
+    "31": "Tomato___Leaf_Mold",
+    "32": "Tomato___Septoria_leaf_spot",
+    "33": "Tomato___Spider_mites Two-spotted_spider_mite",
+    "34": "Tomato___Target_Spot",
+    "35": "Tomato___Tomato_Yellow_Leaf_Curl_Virus",
+    "36": "Tomato___Tomato_mosaic_virus",
+    "37": "Tomato___healthy"
+}

plant_disease_detection.h5

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