rcook commited on
Commit
364aadf
·
verified ·
1 Parent(s): 4451426

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -74,7 +74,7 @@ def summarize():
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"],
@@ -84,7 +84,7 @@ def summarize():
84
  compute_metrics=compute_metrics,
85
  )
86
 
87
- trainer.train()
88
 
89
  return data_collator
90
  # return type(tokenized_billsum)
 
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"],
 
84
  compute_metrics=compute_metrics,
85
  )
86
 
87
+ trainer.train()"""
88
 
89
  return data_collator
90
  # return type(tokenized_billsum)