/predicting_next_word_BERT

Predict the next word using BERT

Primary LanguagePython

next_word_prediction_streamlit

A web app that is capable of predicting next word on it's own.

Table of Content

Demo

Overview

A web application that has a pretained model uncased BERT that will be loaded when they will be selected only once . After that they will serve fast as they are cached. You can select n number of next words that should come after given text.

Installation

  1. Windows user can double click on activation.bat file to install required package
  2. Linux User type following command in commnand line a) First create a virtual environment
python3.7 -m virtualenv venv

b) Move to venv directory and activate environment

cd venv
. bin/activate

c) Clone this project

git clone https://github.com/pandeynandancse/next_word_prediction_streamlit.git

d) Move into cloned directory

cd next_word_prediction_streamlit

e) Now install all requirements

pip install -r requirements.txt

Run

  1. After successfull installation windows user can directly open the link that will be appeared
  2. After successful installation open type
streamlit run next_word.py

and then open link

To Do

  1. Add More Architecture Options
  2. Trigger prediction on each time when space is pressed

Bug / Feature Request

If you find a bug (the website couldn't handle the query and / or gave undesired results), kindly open an issue here by including your search query and the expected result.

If you'd like to request a new function, feel free to do so by opening an issue here. Please include sample queries and their corresponding results.

Contribution

If you'd like to do some contribution, feel free to do so by opening a pull request here. Please include sample queries and their corresponding results.

Technologies Used

Contributor

Nandan Pandey
Nandan Pandey