Update cosine distance

app.py

@@ -137,6 +137,10 @@ def load_code_kmeans_model():
 
 @st.cache_resource(show_spinner="Loading SimilarityCal model...")
 def load_similaritycal_model():
+    """
+    The function to load SimilarityCal model
+    :return: the SimilarityCal model
+    """
     sim_cal_model = PairClassifier()
     sim_cal_model.load_state_dict(torch.load(SIMILARITY_CAL_MODEL_PATH, map_location=device))
     sim_cal_model = sim_cal_model.to(device)
@@ -429,8 +433,8 @@ if __name__ == "__main__":
                     {
                         "name": query_doc.name,
                         "topics": query_doc.topics,
-                        "topic cluster": topic_cluster_number,
                         "code cluster": code_cluster_number,
+                        "topic cluster": topic_cluster_number,
                         "stars": query_doc.stars,
                         "license": query_doc.license,
                     }

data/index_developing.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:18302406f5602e82bddf51bb37223a309f7f93bd518ff252f9ed4331a6d3cc6c
-size 226460129
+oid sha256:3837b4cb3f10cd0ff035201ef44ab655608b2877e5c89efc5cc63a69b666c415
+size 226172318

data/kmeans_model_code_unixcoder.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:bb534645bce9fb19975873003be27e0b386df7550693caed46ee0f1822b16533
-size 967215
+oid sha256:45a9cc8a2e348d81a32ca5fda4e9690b1915d11347f1a343bc4124ae000b38de
+size 500271

data/kmeans_model_topic_scibert.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:48272b4172b3dba079348462044f72f19a004ff65d6cd9222ef424468261f1fb
-size 967215
+oid sha256:36a1852f52907a12cb18fc90fe8b36d3049e123172d9c061c80b4b679685173e
+size 500271