csmith715 commited on
Commit
efce4f7
·
1 Parent(s): 2186f5b

Adding checks

Browse files
Files changed (1) hide show
  1. app.py +11 -1
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