Update main.py

main.py

@@ -61,12 +61,11 @@ def preprocess(query: str, k: int) -> tuple[str, str]:
         tuple[str, str]: A tuple containing the prompt and references
     """
     encoded_query = numpy.expand_dims(model.encode(query), axis=0)
-    print(query, encoded_query)
     faiss.normalize_L2(encoded_query)
     D, I = index.search(encoded_query, k)
     top_five = data.loc[I[0]]
 
-    print(top_five)
+    print(top_five["text"].values)
 
     prompt = (
         "You are an AI assistant who delights in helping people learn about research from the IDETC Conference."