muryshev commited on
Commit
5722846
·
1 Parent(s): 0b23d6c
Files changed (1) hide show
  1. lib/llm_2.py +2 -2
lib/llm_2.py CHANGED
@@ -5,7 +5,7 @@ import json
5
  import pandas as pd
6
 
7
  typedf = pd.read_excel('types.xlsx')
8
- verbose = False
9
  # s = 'Тип обращения: Жалобы > Сектор: Акционерные общества/эмитенты эмиссионных ценных бумаг > Продукт: Дивиденды'
10
  # s = 'Тип обращения: Прочие обращения (запрос, заявление, вопрос) > Сектор: Субъекты страхового дела > Продукт: Прочие обращения (страхование) > Проблема: Иные обращения: Страхование > Обращение может быть типологизировано по иной проблеме: Нет'
11
  level = 'Тип обращения'
@@ -91,7 +91,7 @@ def getCategory(text, categories) :
91
 
92
  def getResponse(prompt) :
93
  url = "https://muryshev-mixtral-api.hf.space/completion"
94
-
95
  payload = json.dumps({
96
  "prompt": '[INST]' + prompt + '[/INST]'
97
  })
 
5
  import pandas as pd
6
 
7
  typedf = pd.read_excel('types.xlsx')
8
+
9
  # s = 'Тип обращения: Жалобы > Сектор: Акционерные общества/эмитенты эмиссионных ценных бумаг > Продукт: Дивиденды'
10
  # s = 'Тип обращения: Прочие обращения (запрос, заявление, вопрос) > Сектор: Субъекты страхового дела > Продукт: Прочие обращения (страхование) > Проблема: Иные обращения: Страхование > Обращение может быть типологизировано по иной проблеме: Нет'
11
  level = 'Тип обращения'
 
91
 
92
  def getResponse(prompt) :
93
  url = "https://muryshev-mixtral-api.hf.space/completion"
94
+ print('Prompt: '+prompt)
95
  payload = json.dumps({
96
  "prompt": '[INST]' + prompt + '[/INST]'
97
  })