RoastMe is a Flask application that utilizes machine learning to generate roasts based on a selfie provided by the user. The application is deployed on Google Cloud Run and is continuously built and deployed using Google Cloud Build and Google Container Registry.
- Accepts a selfie as input and generates a personalized roast
- Trained on 145,000 posts from the subreddit r/RoastMe
- Continuous Deployment through Google Cloud Build
View deployed version here
- Clone the repository
git clone https://github.com/<your-github-username>/roastbot.git
- Navigate to the directory
cd roastbot
- Start the application
python3 main.py
Please note that the roasts generated by the RoastBot may be harsh and offensive in nature. Use at your own discretion.
Contributing Contributions are highly welcomed and appreciated to improve the RoastBot. To contribute, please fork the repository, make your changes and create a pull request.
- Increase the training data from 120,000 to 1 million posts from r/RoastMe for more accurate and relevant results. (Get the roastbot to it's prime)
- Improve the model's performance and accuracy.
- Finally use a combination of an image classifier + LLM model