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)
|