Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
16c3afe
1
Parent(s):
96f677f
Update model.py
Browse files- src/models/model.py +1 -4
src/models/model.py
CHANGED
|
@@ -422,10 +422,7 @@ class Summarization:
|
|
| 422 |
|
| 423 |
self.model = self.model.to(self.device)
|
| 424 |
|
| 425 |
-
def save_model(
|
| 426 |
-
self,
|
| 427 |
-
model_dir="models"
|
| 428 |
-
):
|
| 429 |
"""
|
| 430 |
Save model to dir
|
| 431 |
:param model_dir:
|
|
|
|
| 422 |
|
| 423 |
self.model = self.model.to(self.device)
|
| 424 |
|
| 425 |
+
def save_model(self, model_dir="models"):
|
|
|
|
|
|
|
|
|
|
| 426 |
"""
|
| 427 |
Save model to dir
|
| 428 |
:param model_dir:
|