MicGuest commited on
Commit
e491773
·
unverified ·
1 Parent(s): 2595858

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -7,7 +7,9 @@ model = pipeline(
7
  )
8
 
9
  def predict(prompt):
10
- summary = model(prompt)[0]["summary_text"]
 
 
11
  return summary
12
 
13
 
 
7
  )
8
 
9
  def predict(prompt):
10
+ #summary = model(prompt)[0]["summary_text"]
11
+ summarizer = pipeline(“summarization”, model=”t5-base”, tokenizer=”t5-base”, framework=”tf”)
12
+ summary=summarizer(context, max_length=130, min_length=60)
13
  return summary
14
 
15