SimrusDenuvo commited on
Commit
324dafa
·
verified ·
1 Parent(s): 81cb6db

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +15 -9
app.py CHANGED
@@ -16,12 +16,17 @@ for row in dataset.select(range(3)):
16
 
17
  # Инструкции
18
  cot_instruction = (
19
- "Ты — помощник банка. Клиент задал вопрос. Проанализируй обращение шаг за шагом, "
20
- "выдели ключевые признаки и выдай итоговую категорию обращения."
 
 
21
  )
22
 
23
  simple_instruction = (
24
- "Ты — помощник банка. Определи категорию обращения клиента. Ответ должен быть кратким, без лишнего текста."
 
 
 
25
  )
26
 
27
  # Используемые модели
@@ -42,7 +47,7 @@ def build_cot_prompt(user_input):
42
  def build_simple_prompt(user_input):
43
  examples = "\n\n".join(few_shot_examples)
44
  return (
45
- f"{simple_instruction}\n\n{examples}\n\nКлиент: {user_input}\nКлассификация:"
46
  )
47
 
48
  # Генерация ответов
@@ -95,28 +100,29 @@ def generate_dual_answers(user_input):
95
 
96
  # Gradio интерфейс
97
  with gr.Blocks() as demo:
98
- gr.Markdown("## 🏦 Классификация клиентских обращений — Сравнение моделей и промптов")
99
 
100
- inp = gr.Textbox(label="Вопрос клиента", placeholder="Например: Не приходит СМС-код для входа в приложение", lines=2)
 
101
  btn = gr.Button("Классифицировать")
102
 
103
  gr.Markdown("### ChatGPT-like (ruGPT3small)")
104
  cot1 = gr.Textbox(label="CoT ответ")
105
  cot1_time = gr.Textbox(label="Время CoT")
106
  simple1 = gr.Textbox(label="Zero-shot ответ")
107
- simple1_time = gr.Textbox(label="Время ero-shot")
108
 
109
  gr.Markdown("### GigaChat-like (ruDialoGPT-medium)")
110
  cot2 = gr.Textbox(label="CoT ответ")
111
  cot2_time = gr.Textbox(label="Время CoT")
112
  simple2 = gr.Textbox(label="Zero-shot ответ")
113
- simple2_time = gr.Textbox(label="Время ero-shot")
114
 
115
  gr.Markdown("### DeepSeek-like (RuBERT-tiny2)")
116
  cot3 = gr.Textbox(label="CoT ответ")
117
  cot3_time = gr.Textbox(label="Время CoT")
118
  simple3 = gr.Textbox(label="Zero-shot ответ")
119
- simple3_time = gr.Textbox(label="Время ero-shot")
120
 
121
  btn.click(generate_dual_answers, inputs=[inp], outputs=[
122
  cot1, cot1_time, simple1, simple1_time,
 
16
 
17
  # Инструкции
18
  cot_instruction = (
19
+ "Ты — сотрудник службы безопасности банка. Проанализируй обращение по шагам:\n"
20
+ "1) Определи суть обращения.\n"
21
+ "2) Укажи, есть ли признаки мошенничества (например: звонки, переводы, ссылки, давление).\n"
22
+ "3) Сделай вывод: 'Мошенничество' или 'Не является мошенничеством'.\n\n"
23
  )
24
 
25
  simple_instruction = (
26
+ "Ты — сотрудник службы безопасности банка. Проанализируй обращение по шагам:\n"
27
+ "1) Определи суть обращения.\n"
28
+ "2) Укажи, есть ли признаки мошенничества (например: звонки, переводы, ссылки, давление).\n"
29
+ "3) Сделай вывод: 'Мошенничество' или 'Не является мошенничеством'.\n\n"
30
  )
31
 
32
  # Используемые модели
 
47
  def build_simple_prompt(user_input):
48
  examples = "\n\n".join(few_shot_examples)
49
  return (
50
+ f"{simple_instruction}\n\n{examples}\n\nКлиент: {user_input}\Анализ:"
51
  )
52
 
53
  # Генерация ответов
 
100
 
101
  # Gradio интерфейс
102
  with gr.Blocks() as demo:
103
+ gr.Markdown("## 🛡️ Детектирование мошеннических обращений")
104
 
105
+ inp = gr.Textbox(label="Вопрос клиента", placeholder="Например: Мне позвонили и сказали, что моя карта заблокирована. Попросили ввести номер карты на сайте, чтобы «разблокировать».
106
+ ", lines=2)
107
  btn = gr.Button("Классифицировать")
108
 
109
  gr.Markdown("### ChatGPT-like (ruGPT3small)")
110
  cot1 = gr.Textbox(label="CoT ответ")
111
  cot1_time = gr.Textbox(label="Время CoT")
112
  simple1 = gr.Textbox(label="Zero-shot ответ")
113
+ simple1_time = gr.Textbox(label="Время Zero-shot")
114
 
115
  gr.Markdown("### GigaChat-like (ruDialoGPT-medium)")
116
  cot2 = gr.Textbox(label="CoT ответ")
117
  cot2_time = gr.Textbox(label="Время CoT")
118
  simple2 = gr.Textbox(label="Zero-shot ответ")
119
+ simple2_time = gr.Textbox(label="Время Zero-shot")
120
 
121
  gr.Markdown("### DeepSeek-like (RuBERT-tiny2)")
122
  cot3 = gr.Textbox(label="CoT ответ")
123
  cot3_time = gr.Textbox(label="Время CoT")
124
  simple3 = gr.Textbox(label="Zero-shot ответ")
125
+ simple3_time = gr.Textbox(label="Время Zero-shot")
126
 
127
  btn.click(generate_dual_answers, inputs=[inp], outputs=[
128
  cot1, cot1_time, simple1, simple1_time,