Update app.py

app.py

@@ -40,6 +40,10 @@ messages = [
 prompt = ChatPromptTemplate(messages=messages)
 chain_type_kwargs = {"prompt": prompt}
 
+@cl.author_rename
+def rename(orig_author: str):
+    rename_dict = {"RetrievalQA": "Searching the Rubilab Records"}
+    return rename_dict.get(orig_author, orig_author)
 
 @cl.on_chat_start
 async def init():