sgt444pepper commited on
Commit
1a2c487
·
verified ·
1 Parent(s): 44018ec

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -16,15 +16,14 @@ PROMPT = """/
16
  Не задавай додаткових запитань, а просто запропонуй наявний в контексті товар!
17
  Твоя ціль відповідати на запитання клієнтів, таким чином допомогти їм.
18
  Ти повинен проконсультувати клієнта у виборі товарів використовуючи контекст.
19
- Відповідай тільки українською мовою, назви товарів та компанії залишай англійською.
20
  У разі, якщо конкретну відповідь знайти не вдалося:
21
  - Відповідай "Я не знаю. Для уточнення інформації зверніться за номером: +380954673526" і нічого більше.
22
  Ти завжди дотримуєшся ввічливого, професійного тону. Формат відповіді має бути простим, зрозумілим і чітким. Уникай довгих пояснень, якщо вони не потрібні.
23
 
24
  """
25
 
26
- tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-mpnet-base-v2")
27
- model = AutoModel.from_pretrained("sentence-transformers/all-mpnet-base-v2")
28
  reranker_model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L-6")
29
 
30
  def load_documents(file_paths):
 
16
  Не задавай додаткових запитань, а просто запропонуй наявний в контексті товар!
17
  Твоя ціль відповідати на запитання клієнтів, таким чином допомогти їм.
18
  Ти повинен проконсультувати клієнта у виборі товарів використовуючи контекст.
 
19
  У разі, якщо конкретну відповідь знайти не вдалося:
20
  - Відповідай "Я не знаю. Для уточнення інформації зверніться за номером: +380954673526" і нічого більше.
21
  Ти завжди дотримуєшся ввічливого, професійного тону. Формат відповіді має бути простим, зрозумілим і чітким. Уникай довгих пояснень, якщо вони не потрібні.
22
 
23
  """
24
 
25
+ tokenizer = AutoTokenizer.from_pretrained("sentence-transformers/all-MiniLM-L12-v2")
26
+ model = AutoModel.from_pretrained("sentence-transformers/all-MiniLM-L12-v2")
27
  reranker_model = CrossEncoder("cross-encoder/ms-marco-TinyBERT-L-6")
28
 
29
  def load_documents(file_paths):