hebrew-gpt_neo-small minimal gradio demo using gradio and transformers pipeline

heb_neo_gra_demo.py

import gradio as gr
from transformers import pipeline

title = "Hebrew GPT-Neo Demo"
description = ""
article = "<p></p>"
examples = [
    ['צחוקים ושיגועים'],
    ["למנשה פומפרניקל יש"],
    ["פעם אחת לפני שנים רבות"]
]

model_id = "Norod78/hebrew-gpt_neo-small"
text_generator = pipeline('text-generation', model=model_id, tokenizer=model_id)
max_length = 256
top_k = 40
top_p = 0.92

def text_generation(input_text):
    generated_text = text_generator(input_text,
    max_length=max_length,
    top_k=top_k, 
    top_p=top_p, 
    do_sample=True,
    repetition_penalty=25.0,
    num_return_sequences=1)
    return generated_text[0]["generated_text"]

gr.Interface(
    text_generation,
    [gr.inputs.Textbox(lines=2, label="Enter input text")],
    [gr.outputs.Textbox(type="auto", label="Text Generated")],
    title=title,
    description=description,
    examples=examples,
    theme="default"
).launch()