diff --git a/main.py b/main.py index 0f457b8..8e77298 100644 --- a/main.py +++ b/main.py @@ -187,7 +187,7 @@ async def extract_info( else: raise HTTPException(400, detail="Invalid file type. Please choose 'standard' or 'patent'.") r = requests.post(conf.CHAT_API, json={ - "model": "llama3.1", + "model": conf.CHAT_MODEL, "prompt": prompt, "stream": False }) @@ -195,4 +195,4 @@ async def extract_info( return JSONResponse(content=info) if __name__ == "__main__": - uvicorn.run(app="main:app", host="0.0.0.0", port=8000, reload=True, log_config=LOGGING_CONFIG) \ No newline at end of file + uvicorn.run(app="main:app", host="0.0.0.0", port=2260, reload=True, log_config=LOGGING_CONFIG, debug=False) \ No newline at end of file