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,
|