Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,11 @@ from typing import Optional
|
|
4 |
from transformers import pipeline
|
5 |
|
6 |
app = FastAPI()
|
7 |
-
summarizer = pipeline(
|
|
|
|
|
|
|
|
|
8 |
|
9 |
|
10 |
class InputText(BaseModel):
|
@@ -21,4 +25,4 @@ def greet_json():
|
|
21 |
@app.post("/summary")
|
22 |
def summary(inp: InputText):
|
23 |
print(inp.text)
|
24 |
-
return summarizer(inp.text[:
|
|
|
4 |
from transformers import pipeline
|
5 |
|
6 |
app = FastAPI()
|
7 |
+
summarizer = pipeline(
|
8 |
+
"summarization",
|
9 |
+
# model="Falconsai/text_summarization"
|
10 |
+
model="pszemraj/long-t5-tglobal-base-16384-book-summary"
|
11 |
+
)
|
12 |
|
13 |
|
14 |
class InputText(BaseModel):
|
|
|
25 |
@app.post("/summary")
|
26 |
def summary(inp: InputText):
|
27 |
print(inp.text)
|
28 |
+
return summarizer(inp.text[:3600], max_length=1000, min_length=30, do_sample=False)
|