Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -17,6 +17,7 @@ START = tf.argmax(tf.constant(reserved_tokens) == "[START]")
|
|
| 17 |
END = tf.argmax(tf.constant(reserved_tokens) == "[END]")
|
| 18 |
PAD = tf.argmax(tf.constant(reserved_tokens) == "[PAD]")
|
| 19 |
VOCAB_SIZE = len(vocab)
|
|
|
|
| 20 |
D_MODEL = 256
|
| 21 |
NB_LAYERS = 6
|
| 22 |
FFN_UNITS = 2048
|
|
|
|
| 17 |
END = tf.argmax(tf.constant(reserved_tokens) == "[END]")
|
| 18 |
PAD = tf.argmax(tf.constant(reserved_tokens) == "[PAD]")
|
| 19 |
VOCAB_SIZE = len(vocab)
|
| 20 |
+
print(VOCAB_SIZE)
|
| 21 |
D_MODEL = 256
|
| 22 |
NB_LAYERS = 6
|
| 23 |
FFN_UNITS = 2048
|