Updated the code to limit the api requests

model/api.py

@@ -1,3 +1,8 @@
+from slowapi import Limiter
+from slowapi.util import get_remote_address
+from slowapi.errors import RateLimitExceeded
+from slowapi.responses import JSONResponse
+from starlette.requests import Request
 from fastapi import FastAPI, UploadFile, File
 from fastapi.middleware.cors import CORSMiddleware
 from fastapi.staticfiles import StaticFiles
@@ -28,3 +33,18 @@ async def predict_emotion(file: UploadFile = File(...)):
     image = Image.open(io.BytesIO(contents)).convert("RGB")
     emotion = predict(image)
     return {"emotion": emotion}
+
+limiter = Limiter(key_func=get_remote_address)
+app.state.limiter = limiter
+
+@app.exception_handler(RateLimitExceeded)
+def rate_limit_handler(request: Request, exc: RateLimitExceeded):
+    return JSONResponse(
+        status_code=429,
+        content={"detail": "Too many requests, slow down 😅"}
+    )
+
+@app.post("/api/predict")
+@limiter.limit("5/minute")
+async def predict_emotion(file: UploadFile = File(...)):
+    ...

requirements.txt

@@ -8,4 +8,5 @@ tqdm
 requests
 fastapi
 uvicorn
-python-multipart
+python-multipart
+slowapi