Spaces:
Build error
Build error
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 |
|