Skip to content

Latest commit

 

History

History
59 lines (36 loc) · 1.49 KB

README.md

File metadata and controls

59 lines (36 loc) · 1.49 KB

RAGoji 🕵‍♂️

A tool to generate an emoji based on an emotion, sentence, phrase or even a single word. Implements simple RAG on the llm-emoji-dataset by using MongoDB Atlas Vector search for semantic similarity.

The emoji in the header was generated by RAGoji too!

Usage and Setup

Clone the GitHub repository and navigate to the root folder.

Install the necessary dependencies.

python -m pip install -r requirements.txt

Now run app.py using this command.

uvicorn app:app --reload --port 8000

Open a browser and go to http://localhost:8000/docs to test the endpoint.

CPU Inference

Just install torch for CPU to get started.

python -m pip install torch

Setting up CUDA (Optional)

cuda is recommended for a faster inference if you have a GPU available.

Install the NVIDIA CUDA Toolkit (version greater than or equal to 11.8) and torch >= 11.8.

Note: Make sure the toolkit and torch have same versions (i.e., 11.8)

python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118

Now run the following command in a Python shell to confirm installation.

>> import torch
>> print(torch.cuda.is_available())
True

Acknowledgements

I would like to acknowledge the owner of the llm-emoji-dataset.

Contribution

Feel free to open up a pull request or create an issue.