Spaces:
Runtime error
Runtime error
| import streamlit as st | |
| from transformers import pipeline | |
| def get_generator(): | |
| return pipeline("text-generation", model="sberbank-ai/rugpt3small_based_on_gpt2") | |
| g = get_generator() | |
| st.markdown("**Генератор смешных и не очень текстов**") | |
| input_test = st.text_area("Введите начало текста:") | |
| length = st.slider( | |
| "Отрегeлируйте длину текста:", min_value=20, value=100, max_value=220 | |
| ) | |
| button = st.button("Сгенерировать!") | |
| if button: | |
| res = g( | |
| input_test, | |
| do_sample=True, | |
| max_length=length, | |
| top_k=50, | |
| top_p=0.95, | |
| num_returned_sequences=1, | |
| ) | |
| st.markdown(res[0]["generated_text"]) | |