Spaces:
Sleeping
Sleeping
File size: 422 Bytes
f085c59 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
import streamlit as st
from transformers import pipeline
text_generator = pipeline("text-generation", model="gpt2")
st.title("Hugging Face Transformer Text Generation App")
input_text = st.text_area("Enter text for generation")
if input_text:
generated_text = text_generator(input_text, max_length=100, num_return_sequences=1)[0]['generated_text']
st.subheader("Generated Text:")
st.write(generated_text)
|