3words / app.py
junefisher's picture
Update app.py
45a7010
raw
history blame
215 Bytes
import streamlit as st
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='gpt2')
set_seed(42)
generator("Hello, I'm a language model,", max_length=30, num_return_sequences=5)