app.py

import streamlit as st
import torch
import torchtext

from utils import generate_text, get_model, get_vocab, MAX_LEN, TEMPERATURE

st.title('Dumb Language Model')
st.markdown("Nothing to see here folks. Just a pretty dumb language model. Like seriously, it's pretty dumb.")

text = st.text_input('Enter some text (this will be used to seed the language model)', value='Tell me a story about')
max_len = st.number_input('Enter a max number of words to generate', min_value=0, max_value=512, value=MAX_LEN)
temperature = st.number_input('Enter a temperature (the higher it is, the more random the output will be)', 
                              min_value=0., max_value=100., value=TEMPERATURE)

if st.button('Click to run'):
    vocab = get_vocab()
    model = get_model()
    generated = generate_text(text, model, vocab, max_len=max_len, temperature=temperature)
    
    st.markdown('### Generated Text')
    st.markdown(f'{generated}')