Update app.py

app.py

@@ -2,21 +2,22 @@ import gradio as gr
 from transformers import pipeline
 
 # Загрузка модели для генерации текста
-text_generator = pipeline("text-generation", model="EleutherAI/gpt-neo-2.7B")
+text_generator = pipeline("text-generation")
 
-def chat_bot(input_text):
-    # Генерация ответа на введенный текст
-    response = text_generator(input_text, max_length=100)[0]['generated_text']
-    return response
+# Функция для генерации текста через модель Transformers
+def generate_text(prompt):
+    # Генерация текста на основе введенного пользователем промпта
+    generated_text = text_generator(prompt, max_length=50, do_sample=True)[0]['generated_text']
+    return generated_text
 
 # Создание интерфейса Gradio
-chatbot_interface = gr.Interface(
-    fn=chat_bot,
-    inputs=gr.inputs.Textbox(lines=2, placeholder="Введите ваш вопрос или сообщение..."),
-    outputs=gr.outputs.Textbox(label="Ответ"),
-    title="Чат-бот на основе GPT-Neo",
-    description="Привет! Я готов ответить на ваши вопросы. Просто введите текст и нажмите 'Послать'."
+iface = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    title="Text Generation with Transformers",
+    description="Enter a prompt and let the model generate text based on it."
 )
 
 # Запуск интерфейса
-chatbot_interface.launch()
+iface.launch()