Support AI Pet-Project

Проект для обучения модели T5 на паре «вопрос-ответ» для поддержки такси.

Установка

python -m venv venv
source venv/bin/activate  
deactivate # 
pip install -r requirements.txt

Обучение модели

python train.py # Запуск обучения
python infer.py # Запуск интерактивного режима генерации ответов
Добавление данных:
Файл data/dataset.json содержит пары вопрос-ответ.
Добавляй новые объекты с "input_text" и "target_text".

Пример как работает обученная модель