File size: 514 Bytes
3553dd8 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
import gradio as gr
from transformers import pipeline, set_seed
generator = pipeline('text-generation', model='EleutherAI/gpt-j-6B', 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() |