Update app.py

app.py

@@ -115,7 +115,7 @@ def bot(history, cross_encoder):
     logger.warning('Retrieving documents...')
     
     # if COLBERT RAGATATOUILLE PROCEDURE  : 
-    if cross_encoder=='ColBERT':
+    if cross_encoder=='(HIGH ACCURATE) ColBERT':
         gr.Warning('Retrieving using ColBERT.. First time query will take a minute for model to load..pls wait')
         RAG= RAGPretrainedModel.from_pretrained("colbert-ir/colbertv2.0")
         RAG_db=RAG.from_index('.ragatouille/colbert/indexes/mockingbird')
@@ -150,9 +150,9 @@ def bot(history, cross_encoder):
         logger.warning(f'start cross encoder {len(documents)}')
         # Retrieve documents relevant to query
         query_doc_pair = [[query, doc] for doc in documents]
-        if cross_encoder=='MiniLM-L6v2' :
+        if cross_encoder=='(FAST) MiniLM-L6v2' :
                cross_encoder1 = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2') 
-        elif cross_encoder=='BGE reranker':
+        elif cross_encoder=='(ACCURATE) BGE reranker':
                cross_encoder1 = CrossEncoder('BAAI/bge-reranker-base')
         
         cross_scores = cross_encoder1.predict(query_doc_pair)
@@ -187,7 +187,7 @@ with gr.Blocks(theme='Insuz/SimpleIndigo') as demo:
     gr.HTML(value=f"""
     <p style="font-family: sans-serif; font-size: 16px;">
       A free chat bot assistant for Expenditure Observers on Compendium on Election Expenditure Monitoring using Open source LLMs. <br>
-      The bot can answer questions in natural language, taking relevant extracts from the ECI document which can be accessed <a href="https://www.eci.gov.in/eci-backend/public/api/download?url=LMAhAK6sOPBp%2FNFF0iRfXbEB1EVSLT41NNLRjYNJJP1KivrUxbfqkDatmHy12e%2Fzk1vx4ptJpQsKYHA87guoLjnPUWtHeZgKtEqs%2FyzfTTYIC0newOHHOjl1rl0u3mJBSIq%2Fi7zDsrcP74v%2FKr8UNw%3D%3D" style="color: #FF0000; text-decoration: none;">CLICK HERE !</a>.
+      The bot can answer questions in natural language, taking relevant extracts from the ECI document which can be accessed <a href="https://www.eci.gov.in/eci-backend/public/api/download?url=LMAhAK6sOPBp%2FNFF0iRfXbEB1EVSLT41NNLRjYNJJP1KivrUxbfqkDatmHy12e%2Fzk1vx4ptJpQsKYHA87guoLjnPUWtHeZgKtEqs%2FyzfTTYIC0newOHHOjl1rl0u3mJBSIq%2Fi7zDsrcP74v%2FKr8UNw%3D%3D" style="color: #00008B; text-decoration: none;">CLICK HERE !</a>.
     </p>
     """, elem_id='Sub-heading')
     usage_count = get_and_increment_value_count(db,collection_name, field_name)
@@ -218,7 +218,7 @@ with gr.Blocks(theme='Insuz/SimpleIndigo') as demo:
                 )
         txt_btn = gr.Button(value="Submit text", scale=1)
 
-    cross_encoder = gr.Radio(choices=['MiniLM-L6v2','BGE reranker','ColBERT'], value='BGE reranker',label="Embeddings", info="Choose MiniLM for Speed, BGE reranker for accuracy,ColBERT for HIGH Accuracy (First query to Colbert may take litte time)")
+    cross_encoder = gr.Radio(choices=['(FAST) MiniLM-L6v2','(ACCURATE) BGE reranker','(HIGH ACCURATE) ColBERT'], value='BGE reranker',label="Embeddings", info="Only First query to Colbert may take litte time)")
 
     prompt_html = gr.HTML()
     # Turn off interactivity while generating if you click