rcook commited on
Commit
aba473b
·
verified ·
1 Parent(s): 8de051e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +12 -0
app.py CHANGED
@@ -74,6 +74,18 @@ def summarize():
74
  gradient_accumulation_steps=4 # Effective larger batch size
75
  )
76
 
 
 
 
 
 
 
 
 
 
 
 
 
77
  return data_collator
78
  # return type(tokenized_billsum)
79
 
 
74
  gradient_accumulation_steps=4 # Effective larger batch size
75
  )
76
 
77
+ trainer = Seq2SeqTrainer(
78
+ model=model,
79
+ args=training_args,
80
+ train_dataset=tokenized_billsum["train"],
81
+ eval_dataset=tokenized_billsum["test"],
82
+ processing_class=tokenizer,
83
+ data_collator=data_collator,
84
+ compute_metrics=compute_metrics,
85
+ )
86
+
87
+ trainer.train()
88
+
89
  return data_collator
90
  # return type(tokenized_billsum)
91