Spaces:
Runtime error
Runtime error
quyip
commited on
Commit
·
0433163
1
Parent(s):
d57b505
fix
Browse files- main.py +5 -3
- summary.py +0 -5
main.py
CHANGED
@@ -1,6 +1,8 @@
|
|
1 |
from fastapi import FastAPI
|
2 |
from pydantic import BaseModel
|
|
|
3 |
|
|
|
4 |
app = FastAPI()
|
5 |
|
6 |
|
@@ -16,6 +18,6 @@ class SummaryReq(BaseModel):
|
|
16 |
|
17 |
@app.post("/summary/")
|
18 |
async def summary(request: SummaryReq):
|
19 |
-
|
20 |
-
|
21 |
-
return
|
|
|
1 |
from fastapi import FastAPI
|
2 |
from pydantic import BaseModel
|
3 |
+
from summary import summarize
|
4 |
|
5 |
+
KEY = 'J9l#K4wP5h@2'
|
6 |
app = FastAPI()
|
7 |
|
8 |
|
|
|
18 |
|
19 |
@app.post("/summary/")
|
20 |
async def summary(request: SummaryReq):
|
21 |
+
if request.key != KEY:
|
22 |
+
return 'Unauthorized'
|
23 |
+
return summarize(request.text)
|
summary.py
CHANGED
@@ -17,16 +17,11 @@ def summarize(text: str):
|
|
17 |
return {
|
18 |
"ver": AiSummaryVersion
|
19 |
}
|
20 |
-
print('==============0')
|
21 |
summary = get_summarization(text) if len(text) > 100 else text
|
22 |
-
print('==============1')
|
23 |
translated = get_en_translation(summary)
|
24 |
-
print('==============2')
|
25 |
tags1 = get_classification(translated)
|
26 |
-
print('==============3')
|
27 |
tags2 = get_tags(translated)
|
28 |
tags = filter_tags(list(set(tags1 + tags2)))
|
29 |
-
print('==============4')
|
30 |
|
31 |
return {
|
32 |
"ver": AiSummaryVersion,
|
|
|
17 |
return {
|
18 |
"ver": AiSummaryVersion
|
19 |
}
|
|
|
20 |
summary = get_summarization(text) if len(text) > 100 else text
|
|
|
21 |
translated = get_en_translation(summary)
|
|
|
22 |
tags1 = get_classification(translated)
|
|
|
23 |
tags2 = get_tags(translated)
|
24 |
tags = filter_tags(list(set(tags1 + tags2)))
|
|
|
25 |
|
26 |
return {
|
27 |
"ver": AiSummaryVersion,
|