minor tweaks

app.py

@@ -46,7 +46,7 @@ try:
         
         snapshot_download(
             repo_id=RAG_SOURCE,
-            repo_type="dataset", # Your index is stored as a dataset repo
+            repo_type="dataset", 
             token=hf_token,
             local_dir=LOCAL_DATA_DIR
         )
@@ -57,7 +57,7 @@ try:
     # Load the RAG model from the (now existing) local index path.
     gr.Info(f'''Loading index from {os.path.join(LOCAL_DATA_DIR, "arxiv_colbert")}...''')
     RAG = RAGPretrainedModel.from_index(os.path.join(LOCAL_DATA_DIR, "arxiv_colbert"))
-    RAG.search("Test query", k=1) # Warm-up query
+    _ = RAG.search("Test query", k=1) # Warm-up query
     gr.Info("Retriever loaded successfully!")
 
 except Exception as e: