Spaces:
Sleeping
Sleeping
farrell236
commited on
Commit
·
ab03825
1
Parent(s):
008b649
Update app.py
Browse files
app.py
CHANGED
@@ -36,8 +36,8 @@ def load_model():
|
|
36 |
dff=hparams['dff'],
|
37 |
target_vocab_size=tokenizer.get_vocab_size(),
|
38 |
dropout_rate=hparams['dropout_rate'])
|
39 |
-
transformer.load_weights('checkpoints/
|
40 |
-
print(f'Model Loaded! Checkpoint file: checkpoints/
|
41 |
|
42 |
return transformer, tokenizer
|
43 |
|
|
|
36 |
dff=hparams['dff'],
|
37 |
target_vocab_size=tokenizer.get_vocab_size(),
|
38 |
dropout_rate=hparams['dropout_rate'])
|
39 |
+
transformer.load_weights('checkpoints/RATCHET_202309022247.tf')
|
40 |
+
print(f'Model Loaded! Checkpoint file: checkpoints/RATCHET_202309022247.tf')
|
41 |
|
42 |
return transformer, tokenizer
|
43 |
|