Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -4,7 +4,7 @@ from model import PoemSummaryModel
|
|
4 |
|
5 |
|
6 |
tokenizer = T5TokenizerFast.from_pretrained("t5-base")
|
7 |
-
best_model = PoemSummaryModel.load_from_checkpoint("
|
8 |
best_model.freeze()
|
9 |
|
10 |
|
|
|
4 |
|
5 |
|
6 |
tokenizer = T5TokenizerFast.from_pretrained("t5-base")
|
7 |
+
best_model = PoemSummaryModel.load_from_checkpoint("best-checkpoint.ckpt")
|
8 |
best_model.freeze()
|
9 |
|
10 |
|