asdasdasdasd commited on
Commit
9323d30
·
1 Parent(s): 88468ae

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -2
app.py CHANGED
@@ -28,7 +28,8 @@ def predict(video):
28
  model = K.Sequential([
29
  layers.InputLayer(input_shape=(block_size, 136)),
30
  layers.Dropout(0.25),
31
- layers.Bidirectional(layers.GRU(RNN_UNIT)),
 
32
  layers.Dropout(DROPOUT_RATE),
33
  layers.Dense(64, activation='relu'),
34
  layers.Dropout(DROPOUT_RATE),
@@ -37,7 +38,8 @@ def predict(video):
37
  model_diff = K.Sequential([
38
  layers.InputLayer(input_shape=(block_size - 1, 136)),
39
  layers.Dropout(0.25),
40
- layers.Bidirectional(layers.GRU(RNN_UNIT)),
 
41
  layers.Dropout(DROPOUT_RATE),
42
  layers.Dense(64, activation='relu'),
43
  layers.Dropout(DROPOUT_RATE),
 
28
  model = K.Sequential([
29
  layers.InputLayer(input_shape=(block_size, 136)),
30
  layers.Dropout(0.25),
31
+ #layers.Bidirectional(layers.GRU(RNN_UNIT)),
32
+ keras.layers.wrappers.Bidirectional(keras.layers.recurrent_v2.GRU(RNN_UNIT)),
33
  layers.Dropout(DROPOUT_RATE),
34
  layers.Dense(64, activation='relu'),
35
  layers.Dropout(DROPOUT_RATE),
 
38
  model_diff = K.Sequential([
39
  layers.InputLayer(input_shape=(block_size - 1, 136)),
40
  layers.Dropout(0.25),
41
+ #layers.Bidirectional(layers.GRU(RNN_UNIT)),
42
+ keras.layers.wrappers.Bidirectional(keras.layers.recurrent_v2.GRU(RNN_UNIT)),
43
  layers.Dropout(DROPOUT_RATE),
44
  layers.Dense(64, activation='relu'),
45
  layers.Dropout(DROPOUT_RATE),