Spaces:

AMLGroup4
/

unsupervised_model

Runtime error

ImanAmran commited on Nov 15, 2023

Commit

4f2939f

1 Parent(s): 7b00aa0

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -17,7 +17,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('base.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_128.h5')
 # Database to store embeddings and user IDs
 user_embeddings = []