| import gradio as gr | |
| from transformers import pipeline, set_seed | |
| generator = pipeline('text-generation', model='gpt2', return_full_text=False) | |
| set_seed(42) | |
| def Bemenet(bemenet): | |
| return generator(bemenet, max_length=100, num_return_sequences=2)[1]['generated_text'] | |
| interface = gr.Interface(fn=Bemenet, | |
| title="Cím..", | |
| description="Leírás..", | |
| inputs="text", | |
| outputs="text") | |
| interface.launch() |