This is the official codebase for GastroBot.
GastroBot is a RAG framework Chinese gastric disease chatbot. It uses 25 gastric disease-related guideline documents as a corpus, fine-tuned the embedding model, and is developed based on llama-index. An overview of GastroBot architecture can be seen below:
Set up conda environment and clone the github repo
# create a new environment
$ conda create --name rag python=3.10.13 -y
$ conda activate rag
# install requirements
$ pip install -r requirements.txt
Executing the code in the build_index.ipynb file will build the index locally. The built index is saved in the vector_index folder.
$ streamlit run app.py
Open the browser and enter the address http://127.0.0.1:port/ Start Q&A with it！
- llamaindex: https://www.llamaindex.ai/