Update main.py

main.py

@@ -66,6 +66,8 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
     D, I = index.search(encoded_query, k)
     top_five = data.loc[I[0]]
 
+    print(top_five)
+
     prompt = (
         "You are an AI assistant who delights in helping people learn about research from the IDETC Conference." 
         "Your main task is to provide an ANSWER to the USER_QUERY based on the RESEARCH_EXCERPTS."