gpt2-neo / app.py
PH83's picture
Update app.py
349fe50
raw
history blame contribute delete
385 Bytes
from transformers import pipeline
generator = pipeline('text-generation', model='EleutherAI/gpt-neo-2.7B')
# generator("EleutherAI has", do_sample=True, min_length=50)
def generate(input_text):
return generator(input_text )
iface = gr.Interface(fn = generate,
inputs = 'text',
outputs = 'text',
)
iface.launch()