| import streamlit as st | |
| from models.gpt_MODEL import generate_text | |
| st.title('GPT Text Generation') | |
| prompt_text = st.text_area("Input Text", "Type here...") | |
| length = st.slider("Length of Generated Text", min_value=50, max_value=500, value=200) | |
| temperature = st.slider("Temperature", min_value=0.1, max_value=2.0, value=0.7, step=0.1) | |
| beams = st.slider("Number of Generations", min_value=2, max_value=10, value=4, step=1) | |
| if st.button('Generate Text'): | |
| with st.spinner('Generating...'): | |
| generated_text = generate_text(prompt_text, length, temperature, beams) | |
| st.text_area("Generated Text", generated_text, height=250) |