mmchowdhury commited on
Commit
93ff57e
·
verified ·
1 Parent(s): 9586048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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("checkpoints/best-checkpoint.ckpt")
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