Update app.py
Browse files
app.py
CHANGED
@@ -28,7 +28,7 @@ def summarize_text(text):
|
|
28 |
|
29 |
# Load BART model & tokenizer
|
30 |
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
|
31 |
-
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn")
|
32 |
|
33 |
# Tokenize abstract
|
34 |
inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
|
|
|
28 |
|
29 |
# Load BART model & tokenizer
|
30 |
tokenizer = AutoTokenizer.from_pretrained("facebook/bart-large-cnn")
|
31 |
+
model = AutoModelForSeq2SeqLM.from_pretrained("facebook/bart-large-cnn", low_cpu_mem_usage=True)
|
32 |
|
33 |
# Tokenize abstract
|
34 |
inputs = tokenizer(abstract, return_tensors="pt", truncation=True)
|