Aneeshmishra commited on
Commit
cde183a
·
verified ·
1 Parent(s): f24ee52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ import textwrap, gradio as gr
4
  from transformers import pipeline
5
 
6
  # 1️⃣ small, open model that needs no access-token
7
- MODEL_ID = "sshleifer/distilbart-cnn-12-6"
8
  summariser = pipeline("summarization", model=MODEL_ID, device=-1) # -1 = CPU
9
 
10
  # 2️⃣ rough char limit that maps to the model’s 1 024-token window
 
4
  from transformers import pipeline
5
 
6
  # 1️⃣ small, open model that needs no access-token
7
+ MODEL_ID = "Xenova/distilbart-cnn-6-6"
8
  summariser = pipeline("summarization", model=MODEL_ID, device=-1) # -1 = CPU
9
 
10
  # 2️⃣ rough char limit that maps to the model’s 1 024-token window