varl42 commited on
Commit
9e0447a
·
1 Parent(s): bc1bad1

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("pszemraj/led-base-book-summary")
31
- model = AutoModelForSeq2SeqLM.from_pretrained(model="pszemraj/led-base-book-summary")
32
 
33
  # Tokenize abstract
34
  inputs = tokenizer(abstract,
 
28
 
29
  # Load BART model & tokenizer
30
  tokenizer = AutoTokenizer.from_pretrained("pszemraj/led-base-book-summary")
31
+ model = AutoModelForSeq2SeqLM.from_pretrained("pszemraj/led-base-book-summary")
32
 
33
  # Tokenize abstract
34
  inputs = tokenizer(abstract,