Arjav commited on
Commit
421b057
·
1 Parent(s): b8244d8

Changed tokeniser

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -4,7 +4,7 @@ from transformers import PegasusTokenizer, PegasusForConditionalGeneration
4
 
5
 
6
  def summarize(Terms):
7
- tokenizer = PegasusTokenizer.from_pretrained('google/pegaus-billsum')
8
  model = PegasusForConditionalGeneration.from_pretrained(
9
  "arjav/TOS-Pegasus")
10
  input_tokenized = tokenizer.encode(
 
4
 
5
 
6
  def summarize(Terms):
7
+ tokenizer = PegasusTokenizer.from_pretrained('google/pegasus-billsum')
8
  model = PegasusForConditionalGeneration.from_pretrained(
9
  "arjav/TOS-Pegasus")
10
  input_tokenized = tokenizer.encode(