Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
8327181
1
Parent(s):
b903ee4
fix
Browse files
main.py
CHANGED
@@ -9,7 +9,7 @@ from utils.summary_utils import summarize
|
|
9 |
|
10 |
KEY = 'J9l#K4wP5h@2'
|
11 |
app = FastAPI()
|
12 |
-
executor = ThreadPoolExecutor()
|
13 |
|
14 |
|
15 |
@app.get("/")
|
@@ -28,10 +28,8 @@ async def summary(request: SummaryReq):
|
|
28 |
async def summaries(request: SummariesReq):
|
29 |
if request.key != KEY:
|
30 |
return 'Unauthorized'
|
31 |
-
print(f'process pages length: {len(request.pages)}')
|
32 |
pages_summaries, uncached_pages = get_summarize_from_cache(request.pages)
|
33 |
print(f'processing cached pages: {len(pages_summaries)}, uncached pages: {len(uncached_pages)}')
|
34 |
loop = asyncio.get_event_loop()
|
35 |
await loop.run_in_executor(executor, summarize_un_cache_page, uncached_pages)
|
36 |
-
print('return')
|
37 |
return pages_summaries
|
|
|
9 |
|
10 |
KEY = 'J9l#K4wP5h@2'
|
11 |
app = FastAPI()
|
12 |
+
executor = ThreadPoolExecutor(max_workers=5)
|
13 |
|
14 |
|
15 |
@app.get("/")
|
|
|
28 |
async def summaries(request: SummariesReq):
|
29 |
if request.key != KEY:
|
30 |
return 'Unauthorized'
|
|
|
31 |
pages_summaries, uncached_pages = get_summarize_from_cache(request.pages)
|
32 |
print(f'processing cached pages: {len(pages_summaries)}, uncached pages: {len(uncached_pages)}')
|
33 |
loop = asyncio.get_event_loop()
|
34 |
await loop.run_in_executor(executor, summarize_un_cache_page, uncached_pages)
|
|
|
35 |
return pages_summaries
|