How to: Fast API implementation

[K0nkere]

• FastAPI with pydantic

  
  from fastapi import FastAPI
  from pydantic import BaseModel
  from typing import Optional  

class Img(BaseModel): url: Optional[str] = None class Config: schema_extra = { "example": { "url": "https://test_url.jpg" }
}

app = FastAPI()

@app.get("/ping") def ping(): return {"message": "PONG"}

@app.post("/predict") def predict(image: Img):

pred = model.predict(X)

result = {
  'Healthy': float(pred[0][0]),
  'Rust': float(pred[0][2]),
}

return result