Spaces:
Sleeping
Sleeping
Fix minimum input length
Browse files
app.py
CHANGED
@@ -33,8 +33,8 @@ def train(message: str, epochs: int, learning_rate: float, emb_size: int, inp_le
|
|
33 |
attn_layer = MultiHeadAttention(num_heads=4, key_dim=128)(emb_layer, emb_layer, emb_layer)
|
34 |
noise_layer = GaussianNoise(0.1)(attn_layer)
|
35 |
conv1_layer = Conv1D(64, 8, padding='same', activation='relu', strides=1, input_shape=(64, 128))(noise_layer)
|
36 |
-
conv2_layer = Conv1D(16, 4, padding='
|
37 |
-
conv3_layer = Conv1D(8, 2, padding='
|
38 |
flatten_layer = Flatten()(conv3_layer)
|
39 |
attn_flatten_layer = Flatten()(attn_layer)
|
40 |
conv1_flatten_layer = Flatten()(conv1_layer)
|
|
|
33 |
attn_layer = MultiHeadAttention(num_heads=4, key_dim=128)(emb_layer, emb_layer, emb_layer)
|
34 |
noise_layer = GaussianNoise(0.1)(attn_layer)
|
35 |
conv1_layer = Conv1D(64, 8, padding='same', activation='relu', strides=1, input_shape=(64, 128))(noise_layer)
|
36 |
+
conv2_layer = Conv1D(16, 4, padding='same', activation='relu', strides=1)(conv1_layer)
|
37 |
+
conv3_layer = Conv1D(8, 2, padding='same', activation='relu', strides=1)(conv2_layer)
|
38 |
flatten_layer = Flatten()(conv3_layer)
|
39 |
attn_flatten_layer = Flatten()(attn_layer)
|
40 |
conv1_flatten_layer = Flatten()(conv1_layer)
|