| import streamlit as st | |
| from transformers import pipeline | |
| from datasets import load_dataset | |
| classifier = pipeline("text-generation", model="gpt2-medium") | |
| def main(): | |
| st.title("Text generation") | |
| with st.form("text_field"): | |
| text = st.text_area('enter some text:') | |
| # clicked==True only when the button is clicked | |
| clicked = st.form_submit_button("Submit") | |
| if clicked: | |
| results = classifier([text]) | |
| st.json(results) | |
| if __name__ == "__main__": | |
| main() | |