mirfan commited on
Commit
816d956
·
1 Parent(s): d2d6de0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -130,7 +130,9 @@ def display_UI():
130
  # Call the model with some dummy input data to create its variables
131
  dummy_img = tf.zeros((1, 229, 229, 3))
132
  dummy_seq = tf.zeros((1, 5, 512))
133
- new_model((dummy_img, dummy_seq))
 
 
134
 
135
  # Load the weights
136
  new_model.load_weights('model_weights.h5')
 
130
  # Call the model with some dummy input data to create its variables
131
  dummy_img = tf.zeros((1, 229, 229, 3))
132
  dummy_seq = tf.zeros((1, 5, 512))
133
+ new_model.cnn_model(dummy_img)
134
+ new_model.encoder(dummy_seq)
135
+ new_model.decoder(dummy_seq)
136
 
137
  # Load the weights
138
  new_model.load_weights('model_weights.h5')