Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline | |
# Create a T5 text2text-generation pipeline | |
pipe = pipeline("text2text-generation", model="kaist-ai/prometheus-13b-v1.0") | |
# Define the Streamlit app | |
st.title("Text Generation App") | |
# Ask the user for input text | |
user_input = st.text_area("Enter text:") | |
# Generate text when the user clicks the button | |
if st.button("Generate Text"): | |
if user_input: | |
# Use the specified pipeline to generate text | |
generated_text = pipe(user_input, max_length=100, do_sample=True)[0]["generated_text"] | |
st.write("Generated Text:") | |
st.write(generated_text) | |
else: | |
st.warning("Please enter text.") | |
# Customize the Streamlit app's appearance and other features as needed. | |