Spaces:
Runtime error
Runtime error
| 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) |