Create app.py

app.py

@@ -0,0 +1,42 @@
+from fastapi import FastAPI, UploadFile, File
+from fastapi.responses import JSONResponse
+from ultralytics import YOLO
+import shutil
+import os
+from pathlib import Path
+
+# Load model at startup
+MODEL_PATH = Path("best.pt")
+model = YOLO(str(MODEL_PATH))
+
+app = FastAPI(title="Yellow Disease Detection API")
+
+@app.get("/")
+def root():
+    return {"status": "ok", "message": "YOLOv8 Yellow Disease Detection API is running 🚀"}
+
+@app.post("/predict/")
+async def predict(file: UploadFile = File(...)):
+    # Save temp file
+    tmp_path = f"temp_{file.filename}"
+    with open(tmp_path, "wb") as buffer:
+        shutil.copyfileobj(file.file, buffer)
+
+    # Run YOLO inference
+    results = model.predict(source=tmp_path, conf=0.5, imgsz=640)
+    
+    # Parse predictions
+    preds = []
+    for r in results:
+        for box in r.boxes:
+            preds.append({
+                "cls": int(box.cls.item()),
+                "label": model.names[int(box.cls.item())],
+                "confidence": float(box.conf.item()),
+                "bbox": box.xyxy[0].tolist()
+            })
+
+    # cleanup
+    os.remove(tmp_path)
+
+    return JSONResponse(content={"predictions": preds})