Spaces:

AMLGroup4
/

unsupervised_model

Runtime error

ImanAmran commited on Nov 15, 2023

Commit

1a71f3f

1 Parent(s): afec8c3

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -15,7 +15,7 @@ n_clusters = 5  # You can adjust this based on your data
 kmeans = KMeans(n_clusters=n_clusters)
 # Load the embedding model
-embedding_model = tf.keras.models.load_model('embedding_model.h5')
 # Database to store embeddings and user IDs
 user_embeddings = []

 kmeans = KMeans(n_clusters=n_clusters)
 # Load the embedding model
+embedding_model = tf.keras.models.load_model('base.h5')
 # Database to store embeddings and user IDs
 user_embeddings = []