tcsenpai/pensieve
1
0
Fork 0
mirror of https://github.com/tcsenpai/pensieve.git synced 2025-06-07 03:35:24 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix: ocr fetch bug fix

Browse Source
This commit is contained in:
arkohut 2024-09-11 23:20:42 +08:00
parent dbbc2792ef
commit d98eab9ebd
1 changed files with 2 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
memos/plugins/ocr/main.py
View File

@ -96,12 +96,8 @@ async def predict(img_path):
return None return None
async with httpx.AsyncClient() as client: async with httpx.AsyncClient() as client:
headers = {} headers = {"Authorization": f"Bearer {token}"} if token else {}
if token: return await fetch(endpoint, client, image_base64, headers)
headers["Authorization"] = f"Bearer {token}"
async with semaphore:
ocr_result = await fetch(endpoint, client, image_base64, headers=headers)
return ocr_result
@router.get("/") @router.get("/")