n_gram / app.py
sahiba12's picture
Update app.py
01d172c
raw
history blame contribute delete
729 Bytes
import streamlit as st
from nltk import word_tokenize
from nltk.util import ngrams
import nltk
nltk.download('punkt')
def ngrams_maker(text, n):
tokens = word_tokenize(text)
n_grams = ngrams(tokens, n)
return [' '.join(gram) for gram in n_grams]
def main():
st.title('n_grams maker')
user_input = st.text_area('Enter text:')
n = st.slider('Select n for n-grams:', 2, 5)
if st.button('create'):
if user_input:
st.write(f"*Input Text:* {user_input}")
st.write(f"*{n}-grams Generated:*")
result = ngrams_maker(user_input, n)
st.write(result)
else:
st.warning('Please enter some text.')
if __name__ == '__main__':
main()