import gradio as gr
from transformers import pipeline

# Создаем конвейер генерации текста с помощью предварительно обученной модели
generator = pipeline('text-generation', model='gpt2')

def generate_text(prompt):
    # Генерируем текст на основе введенного пользователем запроса
    result = generator(prompt, max_length=50, num_return_sequences=1)
    return result[0]['generated_text']

# Создаем интерфейс Gradio
interface = gr.Interface(fn=generate_text, 
                         inputs=gr.Textbox(lines=2, placeholder="Введите текст..."), 
                         outputs='text',
                         title="Генератор текста на основе GPT-2",
                         description="Введите начало предложения и получите продолжение от GPT-2.")

# Запускаем приложение
interface.launch()