File size: 385 Bytes
976eb3d 0442431 976eb3d 349fe50 976eb3d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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()
|