jonathanjordan21 commited on
Commit
2d92851
·
verified ·
1 Parent(s): 8a5a3bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -2
app.py CHANGED
@@ -4,7 +4,11 @@ from typing import Optional
4
  from transformers import pipeline
5
 
6
  app = FastAPI()
7
- summarizer = pipeline("summarization", model="Falconsai/text_summarization")
 
 
 
 
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[:1000], max_length=1000, min_length=30, do_sample=False)
 
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)