Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -1,12 +1,15 @@
|
|
1 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
2 |
import torch
|
3 |
import gradio as gr
|
|
|
4 |
|
5 |
# Загрузка модели и токенизатора
|
6 |
tokenizer = AutoTokenizer.from_pretrained("cointegrated/rut5-base-multitask")
|
7 |
model = T5ForConditionalGeneration.from_pretrained("cointegrated/rut5-base-multitask")
|
8 |
|
9 |
-
|
|
|
|
|
10 |
# Формируем чёткий и понятный промт
|
11 |
prompt = """
|
12 |
Описание товара:
|
@@ -44,14 +47,13 @@ model = T5ForConditionalGeneration.from_pretrained("cointegrated/rut5-base-multi
|
|
44 |
description_short.replace('"', "").replace("\\", "")
|
45 |
)
|
46 |
|
47 |
-
|
48 |
# Интерфейс Gradio
|
49 |
demo = gr.Interface(
|
50 |
fn=generate_meta_tags,
|
51 |
inputs=gr.Textbox(label="Введите описание товара", lines=10),
|
52 |
outputs=gr.Textbox(label="Сгенерированные мета-теги"),
|
53 |
title="SEO Meta Tag Generator",
|
54 |
-
description="Генерация
|
55 |
)
|
56 |
|
57 |
if __name__ == "__main__":
|
|
|
1 |
from transformers import AutoTokenizer, T5ForConditionalGeneration
|
2 |
import torch
|
3 |
import gradio as gr
|
4 |
+
import re
|
5 |
|
6 |
# Загрузка модели и токенизатора
|
7 |
tokenizer = AutoTokenizer.from_pretrained("cointegrated/rut5-base-multitask")
|
8 |
model = T5ForConditionalGeneration.from_pretrained("cointegrated/rut5-base-multitask")
|
9 |
|
10 |
+
def generate_meta_tags(description):
|
11 |
+
description = description.strip()
|
12 |
+
|
13 |
# Формируем чёткий и понятный промт
|
14 |
prompt = """
|
15 |
Описание товара:
|
|
|
47 |
description_short.replace('"', "").replace("\\", "")
|
48 |
)
|
49 |
|
|
|
50 |
# Интерфейс Gradio
|
51 |
demo = gr.Interface(
|
52 |
fn=generate_meta_tags,
|
53 |
inputs=gr.Textbox(label="Введите описание товара", lines=10),
|
54 |
outputs=gr.Textbox(label="Сгенерированные мета-теги"),
|
55 |
title="SEO Meta Tag Generator",
|
56 |
+
description="Генерация meta-title и description на основе описания товара"
|
57 |
)
|
58 |
|
59 |
if __name__ == "__main__":
|