Priyanshi3 commited on
Commit
fc9dd81
·
verified ·
1 Parent(s): 759611a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -7,7 +7,7 @@ from dotenv import load_dotenv, find_dotenv
7
  import requests, json
8
  from transformers import pipeline
9
 
10
- get_completion = pipeline("summarization", model="shleifer/distilbart-cnn-12-6")
11
 
12
  def summarize(input):
13
  output = get_completion(input)
 
7
  import requests, json
8
  from transformers import pipeline
9
 
10
+ get_completion = pipeline("summarization", model="sshleifer/distilbart-cnn-12-3")
11
 
12
  def summarize(input):
13
  output = get_completion(input)