File size: 728 Bytes
648aada 260ddae 648aada |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import streamlit as st
from nltk import word_tokenize
from nltk.util import ngrams
nltk.download('punkt')
def generate_ngrams(text, n):
tokens = word_tokenize(text)
n_grams = ngrams(tokens, n)
return [' '.join(gram) for gram in n_grams]
def main():
st.title('N-gram Generator')
user_input = st.text_area('Enter text:')
n = st.slider('Select n for n-grams:', 2, 5)
if st.button('Generate'):
if user_input:
st.write(f"*Input Text:* {user_input}")
st.write(f"*{n}-grams Generated:*")
result = generate_ngrams(user_input, n)
st.write(result)
else:
st.warning('Please enter some text.')
if __name__ == '__main__':
main()
|