Added info for setting up retriever for the first time

app.py

@@ -14,6 +14,12 @@ generate_kwargs = dict(
     )
 
 RAG = RAGPretrainedModel.from_index("colbert/indexes/arxiv_colbert")
+try:
+  gr.Info("Setting up retriever, please wait...")
+  _ = RAG.search("what is Mistral?", k = 1)
+  gr.Info("Retriever working successfully!")
+except:
+  gr.Warning("Retriever not working!")
 
 mark_text = '# 📚 Search Results\n'
 
@@ -58,7 +64,7 @@ with gr.Blocks(theme = gr.themes.Soft()) as demo:
 
     def ask_llm(prompt):
        output = client.text_generation(prompt, **generate_kwargs,  stream=False, details=False, return_full_text=False)
-       output = output.lstrip(' \n']) if output.lstrip().startswith('\n') else output
+       output = output.lstrip(' \n') if output.lstrip().startswith('\n') else output
        return gr.Textbox(output, visible = True)
 
     msg.submit(update_with_rag_md, msg, [gr_md, input]).success(ask_llm, input, output_text)