github/ocr_api_server
5
0
Fork 0
mirror of https://github.com/sml2h3/ocr_api_server.git synced 2025-05-03 23:01:00 +08:00
Code Issues Projects Releases Wiki Activity

修正引用关系

Browse Source
This commit is contained in:
sml2h3 2024-11-13 10:23:50 +08:00
parent 579dbff237
commit dc22b9e823
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
app/main.py
View File

@ -2,8 +2,8 @@ import uvicorn
from fastapi import FastAPI, File, UploadFile, HTTPException, Form from fastapi import FastAPI, File, UploadFile, HTTPException, Form
from typing import Optional, Union from typing import Optional, Union
import base64 import base64
from .models import OCRRequest, SlideMatchRequest, DetectionRequest, APIResponse from app.models import OCRRequest, SlideMatchRequest, DetectionRequest, APIResponse
from .services import ocr_service from app.services import ocr_service
app = FastAPI() app = FastAPI()
@ -38,7 +38,7 @@ async def ocr_endpoint(
png_fix: bool = Form(False) png_fix: bool = Form(False)
): ):
try: try:
if file.size == 0 and image is None: if file is None and image is None:
return APIResponse(code=400, message="Either file or image must be provided") return APIResponse(code=400, message="Either file or image must be provided")
image_bytes = await decode_image(file or image) image_bytes = await decode_image(file or image)
@ -74,7 +74,7 @@ async def detection_endpoint(
image: Optional[str] = Form(None) image: Optional[str] = Form(None)
): ):
try: try:
if file.size == 0 and image is None: if file is None and image is None:
return APIResponse(code=400, message="Either file or image must be provided") return APIResponse(code=400, message="Either file or image must be provided")
image_bytes = await decode_image(file or image) image_bytes = await decode_image(file or image)