style: reformat
This commit is contained in:
parent
3960d106c8
commit
fb20c58fe4
16
src/main.py
16
src/main.py
@ -47,10 +47,12 @@ def root():
|
|||||||
|
|
||||||
return Response(renderTemplate("main.html", {"results": result}))
|
return Response(renderTemplate("main.html", {"results": result}))
|
||||||
|
|
||||||
|
|
||||||
class Image(BaseModel):
|
class Image(BaseModel):
|
||||||
file_name: str
|
file_name: str
|
||||||
file_contents: str
|
file_contents: str
|
||||||
|
|
||||||
|
|
||||||
@app.post("/detect/")
|
@app.post("/detect/")
|
||||||
def file(file: Image):
|
def file(file: Image):
|
||||||
file_name = unquote(file.file_name)
|
file_name = unquote(file.file_name)
|
||||||
@ -72,7 +74,11 @@ def file(file: Image):
|
|||||||
pred_res = predictor(doc)
|
pred_res = predictor(doc)
|
||||||
json_res = pred_res.export()
|
json_res = pred_res.export()
|
||||||
converted = convert_dict_items_to_list(json_res)
|
converted = convert_dict_items_to_list(json_res)
|
||||||
|
|
||||||
|
os.unlink("data/" + file_name)
|
||||||
|
|
||||||
return JSONResponse(content=converted)
|
return JSONResponse(content=converted)
|
||||||
|
|
||||||
except Exception as e:
|
except Exception as e:
|
||||||
raise HTTPException(status_code=422, detail=str(e))
|
raise HTTPException(status_code=422, detail=str(e))
|
||||||
|
|
||||||
@ -81,12 +87,20 @@ def convert_to_list(value):
|
|||||||
if isinstance(value, dict):
|
if isinstance(value, dict):
|
||||||
return {k: convert_to_list(v) for k, v in value.items()}
|
return {k: convert_to_list(v) for k, v in value.items()}
|
||||||
elif isinstance(value, list):
|
elif isinstance(value, list):
|
||||||
return [convert_to_list(item) if isinstance(item, (dict, np.ndarray)) else item.tolist() if isinstance(item, np.ndarray) else item for item in value]
|
return [
|
||||||
|
convert_to_list(item)
|
||||||
|
if isinstance(item, (dict, np.ndarray))
|
||||||
|
else item.tolist()
|
||||||
|
if isinstance(item, np.ndarray)
|
||||||
|
else item
|
||||||
|
for item in value
|
||||||
|
]
|
||||||
elif isinstance(value, np.ndarray):
|
elif isinstance(value, np.ndarray):
|
||||||
return value.tolist()
|
return value.tolist()
|
||||||
else:
|
else:
|
||||||
return value
|
return value
|
||||||
|
|
||||||
|
|
||||||
def convert_dict_items_to_list(d: dict):
|
def convert_dict_items_to_list(d: dict):
|
||||||
converted = {}
|
converted = {}
|
||||||
|
|
||||||
|
Loading…
x
Reference in New Issue
Block a user