Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
8c8cd8b
1
Parent(s):
df766d4
Update model.py
Browse files- 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
|