Adding checks
Browse files
app.py
CHANGED
|
@@ -5,7 +5,7 @@ import numpy as np
|
|
| 5 |
import cv2
|
| 6 |
import uvicorn
|
| 7 |
import PIL.Image as Image
|
| 8 |
-
from fastapi import FastAPI, UploadFile, File, HTTPException
|
| 9 |
from pydantic import BaseModel
|
| 10 |
from ultralytics import YOLO
|
| 11 |
|
|
@@ -116,6 +116,16 @@ async def predict_multipart(file: UploadFile = File(default=None)):
|
|
| 116 |
merged = high | low
|
| 117 |
return PredictResponse(detections=merged)
|
| 118 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 119 |
# @app.post("/predict_base64", response_model=PredictResponse)
|
| 120 |
# def predict_base64(payload: PredictQuery = Body(...)):
|
| 121 |
# b64 = payload.image_base64
|
|
|
|
| 5 |
import cv2
|
| 6 |
import uvicorn
|
| 7 |
import PIL.Image as Image
|
| 8 |
+
from fastapi import FastAPI, UploadFile, File, HTTPException, Request
|
| 9 |
from pydantic import BaseModel
|
| 10 |
from ultralytics import YOLO
|
| 11 |
|
|
|
|
| 116 |
merged = high | low
|
| 117 |
return PredictResponse(detections=merged)
|
| 118 |
|
| 119 |
+
@app.post("/ping")
|
| 120 |
+
async def ping():
|
| 121 |
+
return {"ok": True}
|
| 122 |
+
|
| 123 |
+
@app.post("/echo")
|
| 124 |
+
async def echo(req: Request):
|
| 125 |
+
# echoes JSON or form-data keys without reading big bodies
|
| 126 |
+
ct = req.headers.get("content-type", "")
|
| 127 |
+
return {"ok": True, "content_type": ct}
|
| 128 |
+
|
| 129 |
# @app.post("/predict_base64", response_model=PredictResponse)
|
| 130 |
# def predict_base64(payload: PredictQuery = Body(...)):
|
| 131 |
# b64 = payload.image_base64
|