Update app.py

app.py

@@ -11,8 +11,6 @@ import pickle
 import hdbscan
 
 
-class EuclideanDistance64:
-    pass
 # # Define the prediction function
 def predict_ann(age, workclass, education, occupation, race, gender, capital_gain, capital_loss, hours_per_week, native_country):
     # columns = {
@@ -83,6 +81,8 @@ def predict_hb(age, workclass, education,  occupation,  race, gender, capital_ga
     )
 
     prediction = clusterer.fit_predict(X)
+    filename = 'hdbscan_model.pkl'
+    pickle.dump(clusterer, open(filename, 'wb'))
 
     return f"Predicted Cluster (HDBSCAN): {prediction[-1]}"