jpoptum's picture
Create app.py
cc8cb8e
import streamlit as st
from transformers import pipeline, set_seed
# Define pipeline for text generation
generator = pipeline('text-generation', model='gpt2')
# Set seed for reproducibility
set_seed(42)
# Set up Streamlit app
st.title('Text Generation with Hugging Face')
st.write('Type a prompt below and watch the model generate text!')
# Get user input
prompt = st.text_input('Enter your prompt here:')
# Generate text
if prompt:
text = generator(prompt, max_length=50, num_return_sequences=1)[0]['generated_text']
st.write(text)