Spaces:
Sleeping
Sleeping
Update server1.py
Browse files- server1.py +21 -3
server1.py
CHANGED
|
@@ -8,6 +8,8 @@ from fastapi import FastAPI, File, UploadFile
|
|
| 8 |
from fastapi.responses import JSONResponse
|
| 9 |
from PIL import Image, UnidentifiedImageError
|
| 10 |
import open_clip
|
|
|
|
|
|
|
| 11 |
import os
|
| 12 |
os.environ.setdefault("HF_HOME", "/app/cache")
|
| 13 |
os.environ.setdefault("XDG_CACHE_HOME", "/app/cache")
|
|
@@ -135,13 +137,29 @@ def process_image_bytes(image_bytes: bytes):
|
|
| 135 |
def root():
|
| 136 |
return {"status": "ok", "device": DEVICE}
|
| 137 |
|
|
|
|
| 138 |
@app.post("/predict")
|
| 139 |
-
async def predict(front: UploadFile = File(
|
| 140 |
try:
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 141 |
front_bytes = await front.read()
|
|
|
|
|
|
|
| 142 |
if back is not None:
|
|
|
|
| 143 |
_ = await back.read()
|
|
|
|
| 144 |
result = process_image_bytes(front_bytes)
|
| 145 |
return JSONResponse(content={"code": 200, "data": result})
|
| 146 |
-
|
| 147 |
-
|
|
|
|
|
|
|
|
|
|
|
|
| 8 |
from fastapi.responses import JSONResponse
|
| 9 |
from PIL import Image, UnidentifiedImageError
|
| 10 |
import open_clip
|
| 11 |
+
from fastapi import FastAPI, File, UploadFile, Request
|
| 12 |
+
import traceback
|
| 13 |
import os
|
| 14 |
os.environ.setdefault("HF_HOME", "/app/cache")
|
| 15 |
os.environ.setdefault("XDG_CACHE_HOME", "/app/cache")
|
|
|
|
| 137 |
def root():
|
| 138 |
return {"status": "ok", "device": DEVICE}
|
| 139 |
|
| 140 |
+
|
| 141 |
@app.post("/predict")
|
| 142 |
+
async def predict(front: UploadFile = File(None), back: Optional[UploadFile] = File(None), request: Request = None):
|
| 143 |
try:
|
| 144 |
+
# log de cabeceras y tipos
|
| 145 |
+
if request:
|
| 146 |
+
print("headers:", dict(request.headers))
|
| 147 |
+
if front is None:
|
| 148 |
+
print("no llego 'front'")
|
| 149 |
+
return JSONResponse(content={"code": 400, "error": "faltan archivos: 'front' es obligatorio"}, status_code=200)
|
| 150 |
+
|
| 151 |
+
print("front filename:", front.filename, "content_type:", front.content_type)
|
| 152 |
front_bytes = await front.read()
|
| 153 |
+
print("front size:", len(front_bytes))
|
| 154 |
+
|
| 155 |
if back is not None:
|
| 156 |
+
print("back filename:", back.filename, "content_type:", back.content_type)
|
| 157 |
_ = await back.read()
|
| 158 |
+
|
| 159 |
result = process_image_bytes(front_bytes)
|
| 160 |
return JSONResponse(content={"code": 200, "data": result})
|
| 161 |
+
|
| 162 |
+
except Exception as e:
|
| 163 |
+
print("EXCEPTION:", repr(e))
|
| 164 |
+
traceback.print_exc()
|
| 165 |
+
return JSONResponse(content={"code": 404, "data": {}, "error": str(e)}, status_code=200)
|