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() | |