oOundefinedOo commited on
Commit
6c5fea7
·
verified ·
1 Parent(s): f66be42

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +15 -15
README.md CHANGED
@@ -1,24 +1,24 @@
1
- ---
2
- datasets:
3
- - kuznetsoffandrey/sberquad
4
- language:
5
- - ru
6
- - en
7
- base_model:
8
- - ai-forever/ruT5-base
9
- ---
10
  ## Описание:
11
  Данный проект представляет собой русскоязычную модель для задач "question-answer" (QA), которая генерирует ответы на вопросы, исходя из контекста. В основе модели лежит дообученная версия ruT5-base — трансформера T5, адаптированного для русского языка. Модель была дополнительно обучена на русскоязычном датасете SberQUAD, что позволяет ей эффективно решать задачи извлечения информации из текста.
12
 
13
- ## Пример:
14
  ```python
15
  def generate_answer(question: str, context: str) -> str:
 
 
 
16
 
17
- input_text = f"question: {question} context: {context}"
18
- input_ids = tokenizer.encode(input_text, return_tensors='pt')
19
-
20
- outputs = model.generate(input_ids)
21
- answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
22
 
23
  return answer
24
  ```
 
1
+ ---
2
+ datasets:
3
+ - kuznetsoffandrey/sberquad
4
+ language:
5
+ - ru
6
+ - en
7
+ base_model:
8
+ - ai-forever/ruT5-base
9
+ ---
10
  ## Описание:
11
  Данный проект представляет собой русскоязычную модель для задач "question-answer" (QA), которая генерирует ответы на вопросы, исходя из контекста. В основе модели лежит дообученная версия ruT5-base — трансформера T5, адаптированного для русского языка. Модель была дополнительно обучена на русскоязычном датасете SberQUAD, что позволяет ей эффективно решать задачи извлечения информации из текста.
12
 
13
+ ## Пример использования:
14
  ```python
15
  def generate_answer(question: str, context: str) -> str:
16
+
17
+ input_text = f"question: {question} context: {context}"
18
+ input_ids = tokenizer.encode(input_text, return_tensors='pt')
19
 
20
+ outputs = model.generate(input_ids)
21
+ answer = tokenizer.decode(outputs[0], skip_special_tokens=True)
 
 
 
22
 
23
  return answer
24
  ```