zcdsj
/
cert_recog
5
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

feat: 一些调整

This commit is contained in:
caoqianming 2024-09-24 11:35:23 +08:00
parent 03c467159c
commit 4fd34f943d
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
main.py
View File

@ -187,7 +187,7 @@ async def extract_info(
else: else:
raise HTTPException(400, detail="Invalid file type. Please choose 'standard' or 'patent'.") raise HTTPException(400, detail="Invalid file type. Please choose 'standard' or 'patent'.")
r = requests.post(conf.CHAT_API, json={ r = requests.post(conf.CHAT_API, json={
"model": "llama3.1", "model": conf.CHAT_MODEL,
"prompt": prompt, "prompt": prompt,
"stream": False "stream": False
}) })
@ -195,4 +195,4 @@ async def extract_info(
return JSONResponse(content=info) return JSONResponse(content=info)
if __name__ == "__main__": if __name__ == "__main__":
uvicorn.run(app="main:app", host="0.0.0.0", port=8000, reload=True, log_config=LOGGING_CONFIG) uvicorn.run(app="main:app", host="0.0.0.0", port=2260, reload=True, log_config=LOGGING_CONFIG, debug=False)