ImanAmran commited on
Commit
1a71f3f
·
1 Parent(s): afec8c3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -15,7 +15,7 @@ n_clusters = 5 # You can adjust this based on your data
15
  kmeans = KMeans(n_clusters=n_clusters)
16
 
17
  # Load the embedding model
18
- embedding_model = tf.keras.models.load_model('embedding_model.h5')
19
 
20
  # Database to store embeddings and user IDs
21
  user_embeddings = []
 
15
  kmeans = KMeans(n_clusters=n_clusters)
16
 
17
  # Load the embedding model
18
+ embedding_model = tf.keras.models.load_model('base.h5')
19
 
20
  # Database to store embeddings and user IDs
21
  user_embeddings = []