Moduled Project - это проект на основе Hugging Face Transformers, который включает несколько моделей для генерации текста. В проекте используются такие модели, как GPT-2 и T5. Весь функционал проекта разделен на отдельные модули для удобства использования и расширения.
- Использование моделей GPT-2 и T5 для генерации текста
- Разделение функционала на модули для легкого добавления новых моделей и функций
- Возможность генерировать текст с различными параметрами, такими как
temperature
,num_beams
,weights
и другими - Простой пользовательский интерфейс для взаимодействия с моделями
- Клонируйте репозиторий:
git clone https://github.com/Akilak11/moduled_project.git
- Установите необходимые зависимости из файла
requirements.txt
:
pip install -r requirements.txt
- Загрузите предобученные модели GPT-2 и T5, если их нет локально, и разместите их в каталоге
models/
. Затем обновите пути к моделям в файлахgpt_2.py
иt5.py
. - Запустите
main.py
для работы с проектом:
python main.py
При запуске main.py
вы увидите пользовательский интерфейс, который предлагает выбрать модель и ввести текст для генерации. Выберите одну из доступных моделей, введите начальный текст, и программа сгенерирует продолжение текста на основе выбранной модели и заданных параметров.
Пример ввода:
Выберите модель (gpt-2, t5): gpt-2 Введите начальный текст: Привет, как дела?
Пример вывода:
Привет, как дела? У меня все отлично, спасибо за проявленный интерес! А как у тебя проходит день?
Проект разделен на модули для легкого добавления новых моделей и функций. Вы можете расширить проект, добавляя новые модели или изменяя существующие, а также создавая новые функции или улучшая пользовательский интерфейс.
- Для добавления новых моделей создайте новый файл в каталоге
models/
, например,my_new_model.py
. Затем импортируйте вашу модель вuser_interface.py
и добавьте ее в список доступных моделей. - Если вам нужно добавить новые функции или оптимизировать существующие, вы можете изменять код в соответствующих модулях, а затем обновить
user_interface.py
для интеграции новых функций в пользовательский интерфейс. - Для улучшения пользовательского интерфейса вы можете изменить
user_interface.py
, добавлять новые опции, менять стиль отображения или добавлять новые инструкции для пользователей.
Если у вас возникли вопросы или предложения по улучшению проекта, пожалуйста, создайте issue в репозитории проекта на GitHub или свяжитесь с разработчиком по электронной почте.