eteron's picture
Upload app.py
4421e13 verified
raw
history blame contribute delete
476 Bytes
import gradio as gr
from transformers import pipeline
generator = pipeline('text-generation', model='sberbank-ai/rugpt3large_based_on_gpt2')
def generate(text, history):
result = generator(
text,
do_sample=True, max_length=50, repetition_penalty=5.0,
top_k=5, top_p=0.95, temperature=1,
num_beams=None,
no_repeat_ngram_size=3
)
return result[0]["generated_text"]
demo = gr.ChatInterface(generate)
demo.launch(share=True)