ImanAmran commited on
Commit
2624e1a
·
1 Parent(s): a4c5035

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
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,