An open API service indexing awesome lists of open source software.

https://github.com/hungpm372/retrieval-augmented-generation-app

This is a simple app that demonstrates the use of the retrieval-augmented generation model.
https://github.com/hungpm372/retrieval-augmented-generation-app

chroma langchain openai

Last synced: 5 months ago
JSON representation

This is a simple app that demonstrates the use of the retrieval-augmented generation model.

Awesome Lists containing this project

README

          

# retrieval-augmented-generation-app

This is a simple app that demonstrates the use of the [retrieval-augmented generation](https://arxiv.org/abs/2005.11401) model. The app allows you to enter a query and get a response from the model. The model is fine-tuned on the [Wizard of Wikipedia](https://parl.ai/projects/wizard_of_wikipedia/) dataset.

## Setup
```
pip install poetry
poetry install
```

## Run
```
python main.py
```

## Acknowledgements
- [Hugging Face](https://huggingface.co/)
- [ParlAI](https://parl.ai/)
- [Streamlit](https://streamlit.io/)
- [Wikipedia](https://www.wikipedia.org/)
- [Wizard of Wikipedia](https://parl.ai/projects/wizard_of_wikipedia/)
- [retrieval-augmented generation](https://arxiv.org/abs/2005.11401)
- [Transformers](https://huggingface.co/transformers/)
- [Wikipedia API](https://pypi.org/project/Wikipedia-API/)

## License
This project is licensed under the MIT License - see the [LICENSE](LICENSE) file for details.