varl42 commited on
Commit
1bd3a8d
1 Parent(s): d740958

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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)