Triok1 commited on
Commit
3a75e4a
·
verified ·
1 Parent(s): 8bdbf2c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -3
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__":