Spaces:
Runtime error
Runtime error
| 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) | |