Spaces:
Runtime error
Runtime error
File size: 517 Bytes
43c35de 3338109 57398fb 23c296a 784cc00 406b8e1 784cc00 65ad4aa 784cc00 406b8e1 43c35de | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import streamlit as st
from transformers import pipeline
text_generator = pipeline("text-generation",model="gpt2")
def main():
st.title("Text-generation")
with st.form("text_field"):
text = st.text_area('enter your text:')
# clicked==True only when the button is clicked
clicked = st.form_submit_button("Generate")
if clicked:
results =text_generator(text,max_length=100,num_return_sequences=5)
st.json(results)
if __name__ == "__main__":
main() |