Hugging Face's logo

Spaces:
Arjav
/
TOS-Summarization
Runtime error

App Files Community
2
Arjav commited on
Commit
b8244d8
·
1 Parent(s): 9b77602

Update app.py

Browse files

Changed tokenises

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('nsi319/legal-pegasus')
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/pegaus-billsum')
8
  model = PegasusForConditionalGeneration.from_pretrained(
9
  "arjav/TOS-Pegasus")
10
  input_tokenized = tokenizer.encode(