varl42 commited on
Commit
3e30cd9
·
1 Parent(s): 185001f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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("Falconsai/text_summarization")
31
- model = AutoModelForSeq2SeqLM.from_pretrained("Falconsai/text_summarization")
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)