Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
@@ -10,7 +10,10 @@ from tensorflow.keras import layers, Model
|
|
10 |
def create_embedding_model():
|
11 |
# Load the model architecture
|
12 |
facenet_model = load_model('facenet_keras.h5', compile=False)
|
13 |
-
|
|
|
|
|
|
|
14 |
# Define the embedding model using FaceNet
|
15 |
embedding_model = Model(inputs=facenet_model.input,
|
16 |
outputs=facenet_model.layers[-2].output,
|
|
|
10 |
def create_embedding_model():
|
11 |
# Load the model architecture
|
12 |
facenet_model = load_model('facenet_keras.h5', compile=False)
|
13 |
+
|
14 |
+
# Load the weights from a separate file
|
15 |
+
facenet_model.load_weights('facenet_keras_weights.h5')
|
16 |
+
|
17 |
# Define the embedding model using FaceNet
|
18 |
embedding_model = Model(inputs=facenet_model.input,
|
19 |
outputs=facenet_model.layers[-2].output,
|