ale-dp commited on
Commit
aa21d31
·
verified ·
1 Parent(s): 723b51a

Update Tokenizer

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -47,7 +47,7 @@ summarize_button = col1.button("Summarize")
47
  if summarize_button and user_input:
48
  # Load pre-trained Pegasus model and tokenizer
49
  model_name = "ale-dp/pegasus-finetuned-dialog-summarizer"
50
- tokenizer = AutoTokenizer.from_pretrained(model_name)
51
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
52
 
53
  # Generate summary
 
47
  if summarize_button and user_input:
48
  # Load pre-trained Pegasus model and tokenizer
49
  model_name = "ale-dp/pegasus-finetuned-dialog-summarizer"
50
+ tokenizer = AutoTokenizer.from_pretrained("google/pegasus-cnn_dailymail")
51
  model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
52
 
53
  # Generate summary