Gagan Bhatia commited on
Commit
8c8cd8b
·
1 Parent(s): df766d4

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -0
src/models/model.py CHANGED
@@ -398,6 +398,7 @@ class Summarization:
398
 
399
  def save_model(
400
  self,
 
401
  ):
402
  """
403
  Save model to dir
 
398
 
399
  def save_model(
400
  self,
401
+ model_dir="models"
402
  ):
403
  """
404
  Save model to dir