Create app.py

app.py

@@ -0,0 +1,44 @@
+from fastapi import FastAPI, File, UploadFile
+from fastapi.responses import JSONResponse
+from transformers import pipeline
+from PIL import Image
+import io
+
+# Inisialisasi FastAPI
+app = FastAPI(title="🌿 Plant Disease API (ResNet50)",
+              description="API untuk deteksi penyakit tanaman menggunakan ResNet50 dari Hugging Face.",
+              version="1.0")
+
+# Load model dari Hugging Face
+model = pipeline("image-classification", model="SanketJadhav/PlantDiseaseClassifier-Resnet50")
+
+@app.on_event("startup")
+async def load_model():
+    print("✅ Model ResNet50 siap digunakan (CPU mode).")
+
+@app.get("/")
+async def root():
+    return {"message": "🌿 Plant Disease API is running"}
+
+@app.post("/predict")
+async def predict(file: UploadFile = File(...)):
+    try:
+        # Baca file gambar
+        image_data = await file.read()
+        image = Image.open(io.BytesIO(image_data)).convert("RGB")
+
+        # Prediksi
+        results = model(image)
+
+        # Ambil 3 prediksi teratas
+        top3 = sorted(results, key=lambda x: x['score'], reverse=True)[:3]
+
+        response = {
+            "filename": file.filename,
+            "predictions": top3
+        }
+
+        return JSONResponse(content=response)
+
+    except Exception as e:
+        return JSONResponse(content={"error": str(e)}, status_code=500)