gpt2 / app.py
dengkane's picture
Create app.py
0400dc5
raw
history blame
301 Bytes
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)