Update app.py

app.py

@@ -4,12 +4,7 @@ from transformers import pipeline
 # Загрузка модели для генерации текста
 generator = pipeline("text-generation", model="gpt2")
 
-# Функция для генерации текста по промпту
-def generate_text(prompt):
-    output = generator(prompt, max_length=100)[0]['generated_text']
-    return output
-
-# CSS стили для приложения
+# CSS стили
 css = """
 #generate {
     width: 100%;
@@ -34,9 +29,21 @@ footer {visibility: hidden !important;}
 }
 """
 
+# Функция для генерации текста по prompt
+def generate_text(prompt):
+    output = generator(prompt, max_length=100)[0]['generated_text']
+    return output
+
 # Создание интерфейса Gradio
-with gr.Blocks(css=css) as demo:
-    prompt = gr.inputs.Textbox(lines=5, label="Введите промпт")
-    output = gr.outputs.Textbox(label="Сгенерированный текст")
-    iface = gr.Interface(fn=generate_text, inputs=prompt, outputs=output, title="Hugging Face Text Generator", theme="huggingface")
-    iface.launch()
+demo = gr.Interface(
+    fn=generate_text,
+    inputs="text",
+    outputs="text",
+    layout="vertical",
+    title="Text Generation Demo",
+    css=css
+)
+
+# Запуск интерфейса
+if __name__ == "__main__":
+    demo.launch()