Spaces:

Shamima
/

code-search

Runtime error

Shamima commited on Dec 24, 2022

Commit

648e0b8

•

1 Parent(s): be4eace

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -36,7 +36,7 @@ def get_code(query):
                     truncate="LEFT").embeddings
     # Retrieve the nearest neighbors
-    similar_item_ids = search_index.get_nns_by_vector(query_embed[0],1)
     return data_df.iloc[similar_item_ids[0]]['function_body'], data_df.iloc[similar_item_ids[0]]['file_path']

                     truncate="LEFT").embeddings
     # Retrieve the nearest neighbors
+    similar_item_ids = search_index.get_nns_by_vector(query_embed[0],3)
     return data_df.iloc[similar_item_ids[0]]['function_body'], data_df.iloc[similar_item_ids[0]]['file_path']