Update app.py
Browse files
app.py
CHANGED
@@ -27,8 +27,8 @@ def summarize_text(text):
|
|
27 |
abstract = ". ".join(sentences[start:end+1])
|
28 |
|
29 |
# Load BART model & tokenizer
|
30 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
31 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("
|
32 |
|
33 |
# Tokenize abstract
|
34 |
inputs = tokenizer(abstract, max_length=1024, return_tensors="pt", truncation=True)
|
|
|
27 |
abstract = ". ".join(sentences[start:end+1])
|
28 |
|
29 |
# Load BART model & tokenizer
|
30 |
+
tokenizer = AutoTokenizer.from_pretrained("google/bigbird-pegasus-large-arxiv")
|
31 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("google/bigbird-pegasus-large-arxiv")
|
32 |
|
33 |
# Tokenize abstract
|
34 |
inputs = tokenizer(abstract, max_length=1024, return_tensors="pt", truncation=True)
|