This project aims to develop a Telegram Chatbot based on natural language processing using the BERT model. The model has been trained using movie data from Kaggle and can provide recommendations based on user preferences via Telegram.
In this project, a BERT-based model has been trained to analyze movie details and provide recommendations based on user queries.
Dataset used: Movies Details Dataset
- Python
- BERT
- Natural Language Processing
- Telegram Bot API
- Pandas & NumPy
- Scikit-Learn
- Sentence-Transformers
- NLTK
- Regex
The Movies Details Dataset from Kaggle was used to train the model. The dataset includes the following information:
- Movie Title
- Overview
- Vote Average
Our bot allows users to describe the type and content of the movie they want to watch via Telegram. Based on the provided criteria, the bot lists the most suitable movies and provides recommendations.
pip install numpy pandas nltk sentence-transformers scikit-learn regex python-telegram-bot
Download the Movies Details Dataset and place it in the project directory.
python bert_vectorizer.py
To run your Telegram bot, enter the TOKEN obtained from BotFather in the telegram_bot.py
file.
python telegram_bot.py
User: "Recommend a science fiction movie featuring time travel and parallel universes."
Bot: "See You Yesterday (2019) - Two teenage science prodigies spend their time inventing time machines..."
If you would like to contribute to the project, feel free to submit a pull request or open an issue.