Update app.py
Browse files
app.py
CHANGED
|
@@ -66,7 +66,6 @@ class VectorQuantizer(tf.keras.layers.Layer):
|
|
| 66 |
vq_object = VectorQuantizer(64, 16)
|
| 67 |
embs = np.load('embeddings.npy')
|
| 68 |
vq_object.embeddings = embs
|
| 69 |
-
model = tf.keras.models.load_model('VQ-VAE', custom_objects={'vector_quantizer':vq_object})
|
| 70 |
encoder = model.layers[1]
|
| 71 |
|
| 72 |
#data load and preprocess
|
|
|
|
| 66 |
vq_object = VectorQuantizer(64, 16)
|
| 67 |
embs = np.load('embeddings.npy')
|
| 68 |
vq_object.embeddings = embs
|
|
|
|
| 69 |
encoder = model.layers[1]
|
| 70 |
|
| 71 |
#data load and preprocess
|