from transformers import pipeline, set_seed | |
import streamlit as st | |
x = st.slider('Select a value') | |
st.write(x, 'squared is', x * x) | |
generator = pipeline('text-generation', model='gpt2') | |
set_seed(42) | |
res = generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5) | |
st.write(res) |