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()